pytorch 中的 split


Pytorch中的split問題:

1、使用torch.nn.Conv2d中有個參數是groups會將輸入的feature map分組,此處需要注意的一點是分組之后各組的feature map的channel數目是相同的。

2、另外一種方式用torch.split函數將輸入的feature map先split,之后再卷積出不同輸出channels數目不同的feature maps。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM