详解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简单文本处理的方法
Jul 10 Python
Python中shape计算矩阵的方法示例
Apr 21 Python
详解Django 中是否使用时区的区别
Jun 14 Python
Atom的python插件和常用插件说明
Jul 08 Python
Python实现繁?转为简体的方法示例
Dec 18 Python
Python字符串的常见操作实例小结
Apr 08 Python
python集合是否可变总结
Jun 20 Python
Python中类似于jquery的pyquery库用法分析
Dec 02 Python
使用SimpleITK读取和保存NIfTI/DICOM文件实例
Jul 01 Python
Python selenium爬取微信公众号文章代码详解
Aug 12 Python
python基于openpyxl生成excel文件
Dec 23 Python
python通过opencv调用摄像头操作实例分析
Jun 07 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邮件发送,php发送邮件的类
2011/03/24 PHP
is_uploaded_file函数引发的不能上传文件问题
2013/10/29 PHP
php官方微信接口大全(微信支付、微信红包、微信摇一摇、微信小店)
2015/12/21 PHP
PHP实现的用户注册表单验证功能简单示例
2019/02/25 PHP
IE DOM实现存在的部分问题及解决方法
2009/07/25 Javascript
jQuery封装的获取Url中的Get参数示例
2013/11/26 Javascript
JS版的date函数(和PHP的date函数一样)
2014/05/12 Javascript
JS之相等操作符详解
2016/09/13 Javascript
js 能实现监听F5页面刷新子iframe 而父页面不刷新的方法
2016/11/09 Javascript
JavaScript的for循环中嵌套一个点击事件的问题解决
2017/03/03 Javascript
js实现多张图片延迟加载效果
2017/07/17 Javascript
vue 中swiper的使用教程
2018/05/22 Javascript
vue-quill-editor+plupload富文本编辑器实例详解
2018/10/19 Javascript
javascript异步编程的六种方式总结
2019/05/17 Javascript
Python实现Tab自动补全和历史命令管理的方法
2015/03/12 Python
python 判断网络连通的实现方法
2018/04/22 Python
python实时获取外部程序输出结果的方法
2019/01/12 Python
详解python的四种内置数据结构
2019/03/19 Python
超实用的 30 段 Python 案例
2019/10/10 Python
Pandas操作CSV文件的读写实现方法
2019/11/13 Python
python单元测试框架pytest的使用示例
2020/10/07 Python
多个版本的python共存时使用pip的正确做法
2020/10/26 Python
美国儿童运动鞋和服装零售商:Kids Foot Locker
2017/08/05 全球购物
美国在线购物频道:Shop LC
2019/04/21 全球购物
生产车间班组长岗位职责
2014/01/06 职场文书
《三峡》教学反思
2014/03/01 职场文书
爱与责任演讲稿
2014/05/20 职场文书
供电工程专业求职信
2014/08/09 职场文书
2015年中学元旦晚会活动方案
2014/12/09 职场文书
经理岗位职责
2015/02/02 职场文书
2015年万圣节活动总结
2015/03/24 职场文书
公司奖励通知
2015/04/21 职场文书
银行反洗钱宣传活动总结
2015/05/08 职场文书
暂住证证明
2015/06/19 职场文书
Java循环队列与非循环队列的区别总结
2021/06/22 Java/Android
Python爬虫入门案例之回车桌面壁纸网美女图片采集
2021/10/16 Python