Pytorch 扩展Tensor维度、压缩Tensor维度的方法


Posted in Python onSeptember 09, 2020

1. 扩展Tensor维度

相信刚接触Pytorch的宝宝们,会遇到这样一个问题,输入的数据维度和实验需要维度不一致,输入的可能是2维数据或3维数据,实验需要用到3维或4维数据,那么我们需要扩展这个维度。其实特别简单,只要对数据加一个扩展维度方法就可以了。

1.1torch.unsqueeze(self: Tensor, dim: _int)

torch.unsqueeze(self: Tensor, dim: _int)

参数说明:self:输入的tensor数据,dim:要对哪个维度扩展就输入那个维度的整数,可以输入0,1,2……

1.2Code

第一种方式,输入数据后直接加unsqueeze()

扩展第一维和第二维为1

import torch


def reset_unsqueeze1():
 data = torch.rand([3, 3])
 data1 = data.unsqueeze(dim=0).unsqueeze(dim=1)
 print("data_size: ", data.shape)
 print("data: ", data)
 print("data1_size: ", data1.shape)
 print("data1: ", data1)

结果显示

Pytorch 扩展Tensor维度、压缩Tensor维度的方法

第二种方式,用torch.unsqueeze()

import torch


def reset_unsqueeze2():
 data = torch.rand([3, 3])
 data1 = torch.unsqueeze(data, dim=0)
 print("data_size: ", data.shape)
 print("data: ", data)
 print("data1_size: ", data1.shape)
 print("data1: ", data1)

结果显示

Pytorch 扩展Tensor维度、压缩Tensor维度的方法

2. 压缩Tensor维度

2.1torch.squeeze(self: Tensor, dim: _int)

这个方法刚好和torch.unsqueeze()方法效果相反,压缩Tensor维度。

2.2 Code

第一种方式,输入数据后直接加squeeze()

import torch


def reset_squeeze1():
 data = torch.rand([1, 1, 3, 3])
 data1 = data.squeeze(dim=0).squeeze(dim=1)
 print("data_size: ", data.shape)
 print("data: ", data)
 print("data1_size: ", data1.shape)
 print("data1: ", data1)

结果显示

Pytorch 扩展Tensor维度、压缩Tensor维度的方法

第二种方式,用torch.squeeze()

import torch


def reset_squeeze2():
 data = torch.rand([1, 1, 3, 3])
 data1 = torch.squeeze(data, dim=0)
 print("data_size: ", data.shape)
 print("data: ", data)
 print("data1_size: ", data1.shape)
 print("data1: ", data1)

结果显示

Pytorch 扩展Tensor维度、压缩Tensor维度的方法

到此这篇关于Pytorch 扩展Tensor维度、压缩Tensor维度的方法的文章就介绍到这了,更多相关Pytorch 扩展Tensor维度、压缩Tensor维度内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
教你用Type Hint提高Python程序开发效率
Aug 08 Python
Python实现的基于优先等级分配糖果问题算法示例
Apr 25 Python
Django和Flask框架优缺点对比
Oct 24 Python
Python文件路径名的操作方法
Oct 30 Python
python读取Kafka实例
Dec 23 Python
Pytoch之torchvision.transforms图像变换实例
Dec 30 Python
opencv python如何实现图像二值化
Feb 03 Python
Django权限设置及验证方式
May 13 Python
Python urllib库如何添加headers过程解析
Oct 05 Python
Python实现哲学家就餐问题实例代码
Nov 09 Python
通过代码简单了解django model序列化作用
Nov 12 Python
python安装mysql的依赖包mysql-python操作
Jan 01 Python
Python列表推导式实现代码实例
Sep 09 #Python
利用Python中的Xpath实现一个在线汇率转换器
Sep 09 #Python
pytorch使用horovod多gpu训练的实现
Sep 09 #Python
python,Java,JavaScript实现indexOf
Sep 09 #Python
python 5个顶级异步框架推荐
Sep 09 #Python
python PyAUtoGUI库实现自动化控制鼠标键盘
Sep 09 #Python
Pytorch生成随机数Tensor的方法汇总
Sep 09 #Python
You might like
使用YUI+Ant 实现JS CSS压缩
2014/09/02 PHP
Jquery练习之表单验证实现代码
2010/12/14 Javascript
js控制淡入淡出示例代码
2013/11/12 Javascript
JavaScript实现找出数组中最长的连续数字序列
2014/09/03 Javascript
jQuery获得指定元素坐标的方法
2015/04/14 Javascript
JavaScript框架是什么?怎样才能叫做框架?
2015/07/01 Javascript
jQuery 判断图片是否加载完成方法汇总
2015/08/10 Javascript
IE6-IE9使用JSON、table.innerHTML所引发的问题
2015/12/22 Javascript
jQuery下拉框的简单应用
2016/06/24 Javascript
AngularJS基础 ng-open 指令简单实例
2016/08/02 Javascript
ES6新特性五:Set与Map的数据结构实例分析
2017/04/21 Javascript
vue.js中mint-ui框架的使用方法
2017/05/12 Javascript
JS实现数组去重方法总结(六种方法)
2017/07/14 Javascript
Vue+Flask实现简单的登录验证跳转的示例代码
2018/01/13 Javascript
vue多页面开发和打包正确处理方法
2018/04/20 Javascript
纯js实现无缝滚动功能代码实例
2020/02/21 Javascript
vue实现简单学生信息管理
2020/05/30 Javascript
python实现堆栈与队列的方法
2015/01/15 Python
Python3通过Luhn算法快速验证信用卡卡号的方法
2015/05/14 Python
再谈Python中的字符串与字符编码(推荐)
2016/12/14 Python
Win10下python 2.7与python 3.7双环境安装教程图解
2019/10/12 Python
Django使用Celery加redis执行异步任务的实例内容
2020/02/20 Python
Python try except异常捕获机制原理解析
2020/04/18 Python
摩顿布朗英国官方网上商店:奢华沐浴、身体和头发护理
2016/10/29 全球购物
德国消费电子产品购物网站:Guter Kauf
2020/09/15 全球购物
什么时候用assert
2015/05/08 面试题
有针对性的求职自荐信
2013/11/14 职场文书
小学门卫岗位职责
2013/12/17 职场文书
单位实习证明怎么写
2014/01/17 职场文书
特色蛋糕店创业计划书
2014/01/28 职场文书
项目投资意向书
2014/04/01 职场文书
新兵入伍心得体会
2014/09/04 职场文书
小学中队长竞选稿
2015/11/20 职场文书
《黄道婆》教学反思
2016/02/22 职场文书
台式电脑蓝牙适配器怎么安装?台式电脑蓝牙适配器安装教程
2022/04/08 数码科技
vue数据字典取键值项目的字典问题
2022/04/12 Vue.js