conv2d中的groups参数

groups可以让conv2d分层进行。

比如input的channel为4,如果groups=1, 那么 单个kernel的size就是(4 * kernel size)
如果groups = 2, 单个kernel的size就是(2 * kernel size),那么output至少是2的倍数[如2,4,6,8等]
如果groups = 4, 单个kernel的size为(1 * kernel size),就相当于kernel在逐层(input层)操作,那么output至少是4的倍数。因为卷一次至少也有4个output。

以此类推

你可能感兴趣的:(pytorch,pytorch)