pytorch中的torch.nn.Conv2d()函数图文详解


Posted in Python onFebruary 28, 2022

一、官方文档介绍

官网

pytorch中的torch.nn.Conv2d()函数图文详解

nn.Conv2d:对由多个输入平面组成的输入信号进行二维卷积

pytorch中的torch.nn.Conv2d()函数图文详解

pytorch中的torch.nn.Conv2d()函数图文详解

二、torch.nn.Conv2d()函数详解

参数详解

torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)

参数 参数类型    
in_channels int Number of channels in the input image 输入图像通道数
out_channels int Number of channels produced by the convolution 卷积产生的通道数
kernel_size (int or tuple) Size of the convolving kernel 卷积核尺寸,可以设为1个int型数或者一个(int, int)型的元组。例如(2,3)是高2宽3卷积核
stride (int or tuple, optional) Stride of the convolution. Default: 1 卷积步长,默认为1。可以设为1个int型数或者一个(int, int)型的元组。
padding (int or tuple, optional) Zero-padding added to both sides of the input. Default: 0 填充操作,控制padding_mode的数目。
padding_mode (string, optional) ‘zeros’, ‘reflect’, ‘replicate’ or ‘circular’. Default: ‘zeros’ padding模式,默认为Zero-padding 。
dilation (int or tuple, optional) Spacing between kernel elements. Default: 1 扩张操作:控制kernel点(卷积核点)的间距,默认值:1。
groups (int, optional) Number of blocked connections from input channels to output channels. Default: 1 group参数的作用是控制分组卷积,默认不分组,为1组。
bias (bool, optional) If True, adds a learnable bias to the output. Default: True 为真,则在输出中添加一个可学习的偏差。默认:True。

参数dilation——扩张卷积(也叫空洞卷积)

dilation操作动图演示如下:

Dilated Convolution with a 3 x 3 kernel and dilation rate 2

扩张卷积核为3×3,扩张率为2

pytorch中的torch.nn.Conv2d()函数图文详解

参数groups——分组卷积

Group Convolution顾名思义,则是对输入feature map进行分组,然后每组分别卷积。

pytorch中的torch.nn.Conv2d()函数图文详解

pytorch中的torch.nn.Conv2d()函数图文详解

三、代码实例

import torch

x = torch.randn(3,1,5,4)
print(x)

conv = torch.nn.Conv2d(1,4,(2,3))
res = conv(x)

print(res.shape)    # torch.Size([3, 4, 4, 2])

输入:x[ batch_size, channels, height_1, width_1 ]

  • batch_size,一个batch中样本的个数 3
  • channels,通道数,也就是当前层的深度 1
  • height_1, 图片的高 5
  • width_1, 图片的宽 4

卷积操作:Conv2d[ channels, output, height_2, width_2 ]

  • channels,通道数,和上面保持一致,也就是当前层的深度 1
  • output ,输出的深度 4【需要4个filter】
  • height_2,卷积核的高 2
  • width_2,卷积核的宽 3

输出:res[ batch_size,output, height_3, width_3 ]

  • batch_size,,一个batch中样例的个数,同上 3
  • output, 输出的深度 4
  • height_3, 卷积结果的高度 4
  • width_3,卷积结果的宽度 2

一个样本卷积示例:

pytorch中的torch.nn.Conv2d()函数图文详解

pytorch中的torch.nn.Conv2d()函数图文详解

总结 

到此这篇关于pytorch中torch.nn.Conv2d()函数的文章就介绍到这了,更多相关pytorch torch.nn.Conv2d()函数内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
在Python3中使用asyncio库进行快速数据抓取的教程
Apr 02 Python
Python HTMLParser模块解析html获取url实例
Apr 08 Python
python 实现网上商城,转账,存取款等功能的信用卡系统
Jul 15 Python
Python基于QRCode实现生成二维码的方法【下载,安装,调用等】
Jul 11 Python
浅谈python中的数字类型与处理工具
Aug 02 Python
通过Python 获取Android设备信息的轻量级框架
Dec 18 Python
python正则中最短匹配实现代码
Jan 16 Python
Python实现爬虫从网络上下载文档的实例代码
Jun 13 Python
Django 实现图片上传和显示过程详解
Jul 18 Python
python将字符串list写入excel和txt的实例
Jul 20 Python
在Pytorch中使用样本权重(sample_weight)的正确方法
Aug 17 Python
python多线程爬取西刺代理的示例代码
Jan 30 Python
python3中apply函数和lambda函数的使用详解
Feb 28 #Python
你需要掌握的20个Python常用技巧
Feb 28 #Python
python opencv将多个图放在一个窗口的实例详解
pandas中关于apply+lambda的应用
Feb 28 #Python
Python中的datetime包与time包包和模块详情
Feb 28 #Python
Python 数据可视化神器Pyecharts绘制图像练习
Python使用OpenCV实现虚拟缩放效果
You might like
php发送get、post请求的6种方法简明总结
2014/07/08 PHP
php 使用html5实现多文件上传实例
2016/10/24 PHP
php实现微信模板消息推送
2018/03/30 PHP
PHP开启目录引索+fancyindex漂亮目录浏览带搜索功能
2019/09/23 PHP
PHP语言对接抖音快手小红书视频/图片去水印API接口源码
2020/08/11 PHP
Javascript 获取链接(url)参数的方法
2009/02/15 Javascript
基于jquery实现的定时显示与隐藏div广告的实现代码
2013/08/22 Javascript
有关jquery与DOM节点操作方法和属性记录
2016/04/15 Javascript
详解JavaScript中的事件流和事件处理程序
2016/05/20 Javascript
pc加载更多功能和移动端下拉刷新加载数据
2016/11/07 Javascript
H5移动端图片压缩上传开发流程
2016/11/09 Javascript
js实现右键自定义菜单
2016/12/03 Javascript
简单理解js的冒泡排序
2016/12/19 Javascript
Angular的模块化(代码分享)
2016/12/26 Javascript
解决Mac下安装nmp的淘宝镜像失败问题
2018/05/16 Javascript
微信小程序实现自定义加载图标功能
2018/07/19 Javascript
vue实现表格合并功能
2020/12/01 Vue.js
js实现头像上传并且可预览提交
2020/12/25 Javascript
微信小程序视频弹幕发送功能的实现
2020/12/28 Javascript
[03:26]《DAC最前线》之EG经理自述DOTA2经历
2015/02/02 DOTA
python使用htmllib分析网页内容的方法
2015/05/08 Python
Python基础教程之浅拷贝和深拷贝实例详解
2017/07/15 Python
go和python变量赋值遇到的一个问题
2017/08/31 Python
TensorFlow中权重的随机初始化的方法
2018/02/11 Python
python 接口返回的json字符串实例
2018/03/27 Python
完美解决python中ndarray 默认用科学计数法显示的问题
2018/07/14 Python
python爬虫超时的处理的实例
2018/12/19 Python
由Python编写的MySQL管理工具代码实例
2019/04/09 Python
大专计算机个人求职的自我评价
2013/10/21 职场文书
《珍珠泉》教学反思
2014/02/20 职场文书
清正廉洁演讲稿
2014/05/22 职场文书
电力培训心得体会
2014/09/02 职场文书
项目技术负责人岗位职责
2015/04/13 职场文书
大学入学感言
2015/08/01 职场文书
Nginx Rewrite使用场景及配置方法解析
2021/04/01 Servers
python3.9之你应该知道的新特性详解
2021/04/29 Python