Pytorch——conv2d参数使用

记录下常用参数使用，方便查阅
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是通过公式计算出来的，公式如下：