Pytorch——conv2d参数使用

torch.nn.Conv2d(in_channels, out_channels, kernel_size, 
stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros')

各个参数含义如下图：
输入输出的格式：

input(N,C-in,H-in,W-in)
output(N,C-out,H-out,W-out)

N是batch的大小
C是通道数量
H是输入的高度
W是输入的宽度

其中N和C-in、C-out是人为指定，H，W是原始输入，H-out，W-out是通过公式计算出来的，公式如下：