Group Convolution

m = nn.Conv2d(32, 64, 3, groups=4)
则weight的shape为[64, 8, 3, 3],bias的shape为[64]
可以理解为,输入的通道被划分成了4组,因此每组有8个通道。
输出的通道数为64,即64个kernel,每个kernel的weight的shape为[8, 3, 3],这些kernel被划分成了4组,对应输入的4组。

你可能感兴趣的:(Group Convolution)