Pytorch中的split問題:
1、使用torch.nn.Conv2d中有個參數是groups會將輸入的feature map分組,此處需要注意的一點是分組之后各組的feature map的channel數目是相同的。
2、另外一種方式用torch.split函數將輸入的feature map先split,之后再卷積出不同輸出channels數目不同的feature maps。
Pytorch中的split問題:
1、使用torch.nn.Conv2d中有個參數是groups會將輸入的feature map分組,此處需要注意的一點是分組之后各組的feature map的channel數目是相同的。
2、另外一種方式用torch.split函數將輸入的feature map先split,之后再卷積出不同輸出channels數目不同的feature maps。
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。