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

>>阅读原文<<