X
【batch_size, in_channels, height_1, width_1】
batch_size:一个batch中样本的数量
in_channels:通道数,取决于样本的通道,比如RGB图像的通道数是3
height_1:图片的高
width_1:图片的宽
Conv2d的参数
【channels, number, height_2, width_2】
channels:通道数,要与输入的通道数保持一致,channels=in_channels.
number:卷积核的数量
height_2:卷积核的高
width_2:卷积核的宽
输出:
【batch_size, out_channels, height_3, width_3】
batch_size:一个batch中样本的数量,与输入相同
out_channels:通道数,取决于卷积核的数量,与卷积核的数量number相等
height_3:卷积结果的高
width_3:卷积结果的宽