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