传入神经网络训练是Tensor格式的,图片格式为[b,c,w,h]
b:batch_size 输入的训练数量
c:通道数
w:传入图片的宽
h:传入图片的高
卷积神经网络的计算公式
1、设输入图像尺寸为WH,卷积核大小为FF,步长为S,图像通道数为C,Padding为P,即:
卷积后输出的图像大小为:
2、池化的计算公式
nn.Conv2d( in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True))
设置输入为[64,1,28,28],batch_size=64,C=16
nn.Conv2d(1, 16