Pytorch学习

本文主要是介绍Pytorch学习，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

1. torch.nn.Conv2d

初始化模型：conv = torch.nn.Conv2d(1, 3, kernel_size=3, stride=2, padding=1)

第一个参数：表示输入的通道数
第二个参数：表示输出的通道数
第三个参数：表示卷积核大小
第四个参数：表示步幅长度
第五个参数：是否需要padding

调用模型：out = conv(x)

x是一个四维的tensor：第一维表示图片数量，第二维是通道数（这里一定要跟conv的通道数一致），三四维分别表示图片的宽高
输入也可以将图片reshape成一个一维，conv会自适应的进行计算（将卷积核也reshape成一维）

这篇关于Pytorch学习的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！