Pytorch中Conv2d有下面几个参数: input_channels, output_channels, kernel_size, stride, padding, group 若输入图片为正方形,设输入为(C1, H1, W1),输出为(C2, H2, W2) 则 C2 = out_channels H2 = W2 = ((H1 or W1 - kernel_size + 2 * padding) / stride )+ 1 group参数是分组卷积,input_channels必须要能够被group整除,主要作用是减少网络参数数量加快训练。