pytorch1.0中torch.nn.Conv2d用法详解


Posted in Python onJanuary 10, 2020

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)

以上这篇pytorch1.0中torch.nn.Conv2d用法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
使用python删除nginx缓存文件示例(python文件操作)
Mar 26 Python
python计算圆周长、面积、球体体积并画出圆
Apr 08 Python
利用Python绘制MySQL数据图实现数据可视化
Mar 30 Python
在Docker上开始部署Python应用的教程
Apr 17 Python
Python中的super()方法使用简介
Aug 14 Python
Python实现将MySQL数据库表中的数据导出生成csv格式文件的方法
Jan 11 Python
python将文本中的空格替换为换行的方法
Mar 19 Python
Python 实现网页自动截图的示例讲解
May 17 Python
python提取xml里面的链接源码详解
Oct 15 Python
python找出列表中大于某个阈值的数据段示例
Nov 24 Python
在pytorch中动态调整优化器的学习率方式
Jun 24 Python
浅析PyCharm 的初始设置(知道)
Oct 12 Python
pytorch 利用lstm做mnist手写数字识别分类的实例
Jan 10 #Python
Tensorflow Summary用法学习笔记
Jan 10 #Python
TENSORFLOW变量作用域(VARIABLE SCOPE)
Jan 10 #Python
python numpy数组复制使用实例解析
Jan 10 #Python
关于Pytorch的MNIST数据集的预处理详解
Jan 10 #Python
详解pycharm连接不上mysql数据库的解决办法
Jan 10 #Python
pycharm双击无响应(打不开问题解决办法)
Jan 10 #Python
You might like
windows下升级PHP到5.3.3的过程及注意事项
2010/10/12 PHP
PHP钩子实现方法解析
2019/05/21 PHP
如何在Laravel5.8中正确地应用Repository设计模式
2019/11/26 PHP
PHP数组与字符串互相转换实例
2020/05/05 PHP
Prototype使用指南之base.js
2007/01/10 Javascript
jQuery autocomplete插件修改
2009/04/17 Javascript
jQuery表格插件ParamQuery简单使用方法示例
2013/12/05 Javascript
jQuery调用RESTful WCF示例代码(GET方法/POST方法)
2014/01/26 Javascript
鼠标悬浮停留三秒后自动显示大图js代码
2014/09/09 Javascript
jQuery EasyUI菜单与按钮详解
2016/07/13 Javascript
AngularJS中$watch和$timeout的使用示例
2016/09/20 Javascript
jQuery排序插件tableSorter使用方法
2017/02/10 Javascript
Angular2 http jsonp的实例详解
2017/08/31 Javascript
vue系列之requireJs中引入vue-router的方法
2018/07/18 Javascript
Vue中div contenteditable 的光标定位方法
2018/08/25 Javascript
移动端H5页面返回并刷新页面(BFcache)的方法
2018/11/06 Javascript
通过js示例讲解时间复杂度与空间复杂度
2019/08/06 Javascript
vue 输入电话号码自动按3-4-4分割功能的实现代码
2020/04/30 Javascript
基于Vue CSR的微前端实现方案实践
2020/05/27 Javascript
JavaScript实现原型封装轮播图
2020/12/27 Javascript
跟老齐学Python之print详解
2014/09/28 Python
Python中index()和seek()的用法(详解)
2017/04/27 Python
python实现对指定字符串补足固定长度倍数截断输出的方法
2018/11/15 Python
学习python的前途 python挣钱
2019/02/27 Python
python三方库之requests的快速上手
2019/03/04 Python
pyqt5 实现 下拉菜单 + 打开文件的示例代码
2019/06/20 Python
使用Windows批处理和WMI设置Python的环境变量方法
2019/08/14 Python
CSS3 对过渡(transition)进行调速以及延时
2020/10/21 HTML / CSS
HTML5 和小程序实现拍照图片旋转、压缩和上传功能
2018/10/08 HTML / CSS
本科生职业生涯规划书范文
2014/01/21 职场文书
招标授权委托书样本
2014/09/23 职场文书
党的群众路线教育实践活动心得体会范文
2014/11/05 职场文书
地方课程教学计划
2015/01/19 职场文书
mysql 联合索引生效的条件及索引失效的条件
2021/11/20 MySQL
20180830晚上第一届KSL半决赛 雨神vs解冻(二龙 三炮解说)
2022/04/01 星际争霸
React如何使用axios请求数据并把数据渲染到组件
2022/08/05 Javascript