卷积操作

nn.Conv2d(input_cahnnel, output_channel, conv_szie)

输入图像:
pytorch学习:基础卷积运算
卷积核的设置:
pytorch学习:基础卷积运算输入的维度是1,输出维度是2,卷积核大小3×3

先对图像的格式变化成NCHW:
pytorch学习:基础卷积运算

卷积之后的结果:
pytorch学习:基础卷积运算

relu之后的结果:

pytorch学习:基础卷积运算max_pool之后的结果:
pytorch学习:基础卷积运算

参考资源:
60题目pytorch

相关文章: