pytorch的conv2d函數groups分組卷積使用及理解

重點在於groups參數的理解 1、groups參數是如何影響卷積計算的 conv = nn.Conv2d(in_channels=6, out_channels=6, kernel_size=1, groups=3) conv.weight.data.size() 輸出： torch.Size([6, 2, 1, 1]) （此時轉置參數Transposed默認爲False，源碼如下) 當grou

>>阅读原文<<