详解pytorch中squeeze()和unsqueeze()函数介绍


Posted in Python onSeptember 03, 2020

squeeze的用法主要就是对数据的维度进行压缩或者解压。

先看torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的数去掉第一个维数为一的维度之后就变成(3)行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=torch.squeeze(a,N) a中去掉指定的定的维数为一的维度。

再看torch.unsqueeze()这个函数主要是对数据维度进行扩充。给指定位置加上维数为一的维度,比如原本有个三行的数据(3),在0的位置加了一维就变成一行三列(1,3)。a.squeeze(N) 就是在a中指定位置N加上一个维数为1的维度。还有一种形式就是b=torch.squeeze(a,N) a就是在a中指定位置N加上一个维数为1的维度

一、unsqueeze()函数

1. 首先初始化一个a

详解pytorch中squeeze()和unsqueeze()函数介绍

可以看出a的维度为(2,3)

2. 在第二维增加一个维度,使其维度变为(2,1,3)

详解pytorch中squeeze()和unsqueeze()函数介绍

可以看出a的维度已经变为(2,1,3)了,同样如果需要在倒数第二个维度上增加一个维度,那么使用b.unsqueeze(-2)

二、squeeze()函数介绍

1. 首先得到一个维度为(1,2,3)的tensor(张量)

详解pytorch中squeeze()和unsqueeze()函数介绍

由图中可以看出c的维度为(1,2,3)

2.下面使用squeeze()函数将第一维去掉

详解pytorch中squeeze()和unsqueeze()函数介绍

可见,维度已经变为(2,3)

3.另外

详解pytorch中squeeze()和unsqueeze()函数介绍

可以看出维度并没有变化,仍然为(1,2,3),这是因为只有维度为1时才会去掉。

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

Python 相关文章推荐
Python open读写文件实现脚本
Sep 06 Python
Python入门篇之对象类型
Oct 17 Python
深入浅析python定时杀进程
Jun 06 Python
windows 下python+numpy安装实用教程
Dec 23 Python
python+selenium实现163邮箱自动登陆的方法
Dec 31 Python
Python实现基于TCP UDP协议的IPv4 IPv6模式客户端和服务端功能示例
Mar 22 Python
python和shell获取文本内容的方法
Jun 05 Python
通过python改变图片特定区域的颜色详解
Jul 15 Python
在自动化中用python实现键盘操作的方法详解
Jul 19 Python
Python使用APScheduler实现定时任务过程解析
Sep 11 Python
Python实现直播推流效果
Nov 26 Python
解决Python import docx出错DLL load failed的问题
Feb 13 Python
Django实现微信小程序支付的示例代码
Sep 03 #Python
解决Python3.7.0 SSL低版本导致Pip无法使用问题
Sep 03 #Python
一篇文章搞懂python的转义字符及用法
Sep 03 #Python
容易被忽略的Python内置类型
Sep 03 #Python
python类共享变量操作
Sep 03 #Python
Python pip install之SSL异常处理操作
Sep 03 #Python
详解python tkinter 图片插入问题
Sep 03 #Python
You might like
php封装好的人民币数值转中文大写类
2015/12/20 PHP
PHPMailer ThinkPHP实现自动发送邮件功能
2018/06/10 PHP
JavaScript语言中的Literal Syntax特性分析
2007/03/08 Javascript
Javascript模板技术
2007/04/27 Javascript
JavaScript语法着色引擎(demo及打包文件下载)
2007/06/13 Javascript
Ext 表单布局实例代码
2009/04/30 Javascript
Javascript中常见的校验如域名、手机、邮箱等等
2014/01/02 Javascript
jQuery网页选项卡插件rTabs用法实例分析
2015/08/26 Javascript
jquery ztree实现模糊搜索功能
2016/02/25 Javascript
jQuery对象的链式操作用法分析
2016/05/10 Javascript
JS冒泡事件与事件捕获实例详解
2016/11/25 Javascript
webpack结合express实现自动刷新的方法
2019/05/07 Javascript
jquery实现轮播图特效
2020/04/12 jQuery
原生JS实现无缝轮播图片
2020/06/24 Javascript
[00:50]2014DOTA2国际邀请赛 NEWBEE战队回顾
2014/08/01 DOTA
[43:35]EG vs Winstrike 2018国际邀请赛小组赛BO2 第一场 8.18
2018/08/19 DOTA
python读取二进制mnist实例详解
2017/05/31 Python
人机交互程序 python实现人机对话
2017/11/14 Python
Python实现批量压缩图片
2018/01/25 Python
Python的argparse库使用详解
2018/10/09 Python
Python+selenium 获取浏览器窗口坐标、句柄的方法
2018/10/14 Python
python爬虫神器Pyppeteer入门及使用
2019/07/13 Python
python3 dict ndarray 存成json,并保留原数据精度的实例
2019/12/06 Python
在服务器上安装python3.8.2环境的教程详解
2020/04/26 Python
The Hut德国站点:时装、家居用品、美容等
2016/09/23 全球购物
保护环境建议书300字
2014/05/13 职场文书
大学生入党积极分子自我评价
2014/09/20 职场文书
师德师风剖析材料
2014/09/30 职场文书
2015年宣传思想工作总结
2015/05/22 职场文书
有关朝花夕拾的读书笔记
2015/06/29 职场文书
新课程改革心得体会
2016/01/22 职场文书
怎样做好公众演讲能力?
2019/08/28 职场文书
Python爬虫之爬取某文库文档数据
2021/04/21 Python
面试被问select......for update会锁表还是锁行
2021/11/11 MySQL
spring cloud eureka 服务启动失败的原因分析及解决方法
2022/03/17 Java/Android
Python装饰器详细介绍
2022/03/25 Python