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 相关文章推荐
举例讲解Python中的迭代器、生成器与列表解析用法
Mar 20 Python
巧用python和libnmapd,提取Nmap扫描结果
Aug 23 Python
python数据类型_字符串常用操作(详解)
May 30 Python
Django 多环境配置详解
May 14 Python
python flask中动态URL规则详解
Nov 22 Python
python实现简单学生信息管理系统
Apr 09 Python
利用pyecharts读取csv并进行数据统计可视化的实现
Apr 17 Python
解决Jupyter Notebook使用parser.parse_args出现错误问题
Apr 20 Python
Python基于yaml文件配置logging日志过程解析
Jun 23 Python
python 贪心算法的实现
Sep 18 Python
python time.strptime格式化实例详解
Feb 03 Python
浅谈pytorch中的dropout的概率p
May 27 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
PHP中file_exists函数不支持中文名的解决方法
2014/07/26 PHP
Laravel 4 初级教程之视图、命名空间、路由
2014/10/30 PHP
php实现字符串首字母转换成大写的方法
2015/03/17 PHP
php安装ssh2扩展的方法【Linux平台】
2016/07/20 PHP
php实现批量修改文件名称的方法
2016/07/23 PHP
php对微信支付回调处理的方法
2018/08/23 PHP
php session_decode函数用法讲解
2019/05/26 PHP
php桥接模式应用案例分析
2019/10/23 PHP
把textarea中字符串里含有的回车换行替换成<br>的javascript代码
2007/04/20 Javascript
Jquery 获取checkbox的checked问题
2011/11/16 Javascript
jQuery ajax调用WCF服务实例
2014/07/16 Javascript
javascript实现根据身份证号读取相关信息
2014/12/17 Javascript
浅谈JSON中stringify 函数、toJosn函数和parse函数
2015/01/26 Javascript
Javascript无参数和有参数类继承问题解决方法
2015/03/02 Javascript
js实现宇宙星空背景效果的方法
2015/03/03 Javascript
javascript实现动态改变层大小的方法
2015/05/14 Javascript
js实现的二级横向菜单条实例
2015/08/22 Javascript
Sea.JS知识总结
2016/05/05 Javascript
浅谈JQuery+ajax+jsonp 跨域访问
2016/06/25 Javascript
JS实现获取来自百度,Google,soso,sogou关键词的方法
2016/12/21 Javascript
JS正则表达式验证密码格式的集中情况总结
2017/02/23 Javascript
Angular 4.x+Ionic3踩坑之Ionic 3.x界面传值详解
2018/03/13 Javascript
详解Vue webapp项目通过HBulider打包原生APP(vue+webpack+HBulider)
2019/02/02 Javascript
Node.js使用MongoDB的ObjectId作为查询条件的方法
2019/09/10 Javascript
详解element上传组件before-remove钩子问题解决
2020/04/08 Javascript
[03:07]完美世界DOTA2联赛PWL DAY10 决赛集锦
2020/11/11 DOTA
Python functools模块学习总结
2015/05/09 Python
Python删除windows垃圾文件的方法
2015/07/14 Python
python在不同层级目录import模块的方法
2016/01/31 Python
pycharm远程开发项目的实现步骤
2019/01/20 Python
总结python中pass的作用
2019/02/27 Python
python项目对接钉钉SDK的实现
2019/07/15 Python
python 如何在测试中使用 Mock
2021/03/01 Python
工程部主管岗位职责
2013/11/17 职场文书
护理专业求职信
2014/06/15 职场文书
Oracle笔记
2021/04/05 Oracle