C/C++教程

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学习的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!