详解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 struct.unpack
Sep 06 Python
多线程爬虫批量下载pcgame图片url 保存为xml的实现代码
Jan 17 Python
Python两个整数相除得到浮点数值的方法
Mar 18 Python
Django学习笔记之Class-Based-View
Feb 15 Python
python 容器总结整理
Apr 04 Python
机器学习经典算法-logistic回归代码详解
Dec 22 Python
python 把文件中的每一行以数组的元素放入数组中的方法
Apr 29 Python
Python3.5基础之变量、数据结构、条件和循环语句、break与continue语句实例详解
Apr 26 Python
Python中使用filter过滤列表的一个小技巧分享
May 02 Python
Python 如何调试程序崩溃错误
Aug 03 Python
Python 通过爬虫实现GitHub网页的模拟登录的示例代码
Aug 17 Python
python程序的组织结构详解
Dec 06 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匹配UTF-8中文的正则表达式
2015/05/13 PHP
CentOS7.0下安装PHP5.6.30服务的教程详解
2018/09/29 PHP
PHP+iframe模拟Ajax上传文件功能示例
2019/07/02 PHP
tp5框架的增删改查操作示例
2019/10/31 PHP
jQuery判断元素是否是隐藏的代码
2011/04/24 Javascript
推荐40款强大的 jQuery 导航插件和教程(上篇)
2012/09/14 Javascript
单击和双击事件的冲突处理示例代码
2014/04/03 Javascript
js实现双击图片放大单击缩小的方法
2015/02/17 Javascript
jquery显示loading图片直到网页加载完成的方法
2015/06/25 Javascript
JavaScript取得键盘按下方向键是哪个的方法
2015/08/04 Javascript
你不知道的高性能JAVASCRIPT
2016/01/18 Javascript
Angularjs全局变量被作用域监听的正确姿势
2016/02/06 Javascript
AngularJS整合Springmvc、Spring、Mybatis搭建开发环境
2016/02/25 Javascript
easyui 中的datagrid跨页勾选问题的实现方法
2017/01/18 Javascript
vue的基本用法与常见指令
2017/08/15 Javascript
[02:51]2018年度DOTA2最佳中单位选手-完美盛典
2018/12/17 DOTA
[04:50]2019DOTA2高校联赛秋季赛四强集锦
2019/12/27 DOTA
[45:44]完美世界DOTA2联赛PWL S2 FTD vs PXG 第一场 11.27
2020/12/01 DOTA
pip 错误unused-command-line-argument-hard-error-in-future解决办法
2014/06/01 Python
Python类的多重继承问题深入分析
2014/11/09 Python
python实现数据库跨服务器迁移
2018/04/12 Python
基于Python 装饰器装饰类中的方法实例
2018/04/21 Python
详解pandas库pd.read_excel操作读取excel文件参数整理与实例
2019/02/17 Python
Pandas库之DataFrame使用的学习笔记
2019/06/21 Python
PHP统计代码行数的小代码
2019/09/19 Python
Python 如何优雅的将数字转化为时间格式的方法
2019/09/26 Python
Python selenium页面加载慢超时的解决方案
2020/03/18 Python
Python中读取文件名中的数字的实例详解
2020/12/25 Python
Marriott国际:万豪国际酒店查询预订
2017/09/25 全球购物
新西兰廉价汽车租赁:Snap Rentals
2018/09/14 全球购物
Java多态性的定义以及类型
2014/09/16 面试题
电子商务助理求职自荐信
2014/04/10 职场文书
餐饮商业计划书范文
2014/04/29 职场文书
公司授权委托书样本
2014/09/15 职场文书
四风问题对照检查材料整改措施
2014/09/27 职场文书
Vue中Object.assign清空数据报错的解决方案
2022/03/03 Vue.js