最新下载
热门教程
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
pytorch1.0中torch.nn.Conv2d如何用 pytorch1.0中torch.nn.Conv2d用法解析
时间:2020-01-10 编辑:袖梨 来源:一聚教程网
本篇文章小编给大家分享一下pytorch1.0中torch.nn.Conv2d用法解析,小编觉得挺不错的,现在分享给大家供大家参考,有需要的小伙伴们可以来看看。
Conv2d的简单使用
torch 包 nn 中 Conv2d 的用法与 tensorflow 中类似,但不完全一样。
在 torch 中,Conv2d 有几个基本的参数,分别是
in_channels 输入图像的深度
out_channels 输出图像的深度
kernel_size 卷积核大小,正方形卷积只为单个数字
stride 卷积步长,默认为1
padding 卷积是否造成尺寸丢失,1为不丢失
与tensorflow不一样的是,pytorch中的使用更加清晰化,我们可以使用这种方法定义输入与输出图像的深度并同时指定使用的卷积核的大小。
而我们的输入则经由 Conv2d 定义的参数传入,如下所示:
# 定义一个输入深度为1,输出为6,卷积核大小为 3*3 的 conv1 变量 self.conv1 = nn.Conv2d(1, 6, 3) # 传入原始输入x,以获得长宽与x相当,深度为6的卷积部分 x = self.conv1(x)
要注意的是,Conv2d中所需要的输入顺序为
batchsize, nChannels, Height, Width
其他的简单使用
同样的,与 Conv2d 类似的函数还有很多,类似 max_pool2d、relu等等,他们的使用方法与 Conv2d 极为类似,如下所示:
# relu函数的使用 F.relu(self.conv1(x)) # 池化函数的使用 F.max_pool2d(F.relu(self.conv2(x)), 2)
相关文章
- Golang ProtoBuf的基本语法详解 10-20
- Python识别MySQL中的冗余索引解析 10-20
- Python+Pygame绘制小球代码展示 10-18
- Python中的数据精度问题介绍 10-18
- Python随机值生成的常用方法介绍 10-18
- python3解压缩.gz文件分析 09-27