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切片原理和方法
Dec 19 Python
一个月入门Python爬虫学习,轻松爬取大规模数据
Jan 03 Python
微信跳一跳游戏python脚本
Apr 01 Python
python实现两个文件合并功能
Apr 01 Python
python实现对文件中图片生成带标签的txt文件方法
Apr 27 Python
利用python的socket发送http(s)请求方法示例
May 07 Python
Python3 单行多行万能正则匹配方法
Jan 07 Python
Gauss-Seidel迭代算法的Python实现详解
Jun 29 Python
用Python实现最速下降法求极值的方法
Jul 10 Python
Django 创建/删除用户的示例代码
Jul 24 Python
Python绘图实现显示中文
Dec 04 Python
Python如何操作docker redis过程解析
Aug 10 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
dvwa+xampp搭建显示乱码的问题及解决方案
2015/08/23 PHP
PHP实现的随机IP函数【国内IP段】
2016/07/20 PHP
php 计算两个时间相差的天数、小时数、分钟数、秒数详解及实例代码
2016/11/09 PHP
php + WebUploader实现图片批量上传功能
2019/05/06 PHP
mouse_on_title.js
2006/08/25 Javascript
不一样的文字闪烁 轮番闪烁
2009/11/11 Javascript
Javascript attachEvent传递参数的办法
2009/12/14 Javascript
javascript实现单击和双击并存的方法
2014/12/13 Javascript
详谈javascript中的cookie
2015/06/03 Javascript
jQuery判断checkbox选中状态
2016/05/12 Javascript
webpack+vue.js实现组件化详解
2016/10/12 Javascript
windows 下安装nodejs 环境变量设置
2017/02/02 NodeJs
Node连接mysql数据库方法介绍
2017/02/07 Javascript
ES6新特性:使用export和import实现模块化详解
2017/07/31 Javascript
10行原生JS实现文字无缝滚动(超简单)
2018/01/02 Javascript
vue scroller返回页面记住滚动位置的实例代码
2018/01/29 Javascript
Fetch超时设置与终止请求详解
2019/05/18 Javascript
[02:42]完美大师赛主赛事淘汰赛第三日观众采访
2017/11/25 DOTA
python中解析json格式文件的方法示例
2017/05/03 Python
网站渗透常用Python小脚本查询同ip网站
2017/05/08 Python
详谈python read readline readlines的区别
2017/09/22 Python
python库lxml在linux和WIN系统下的安装
2018/06/24 Python
Python 做曲线拟合和求积分的方法
2018/12/29 Python
python并发编程多进程 互斥锁原理解析
2019/08/20 Python
python pprint模块中print()和pprint()两者的区别
2020/02/10 Python
Python中的全局变量如何理解
2020/06/04 Python
CSS3教程(9):设置RGB颜色
2009/04/02 HTML / CSS
法国珠宝店:CLEOR
2017/01/29 全球购物
聪明的粉丝购买门票的地方:TickPick
2018/03/09 全球购物
怎样写好创业计划书的内容
2014/02/06 职场文书
护理毕业生自我鉴定
2014/02/11 职场文书
2015年扶贫帮困工作总结
2015/05/20 职场文书
2016中学教师读书心得体会
2016/01/13 职场文书
创业计划书之美甲店
2019/09/20 职场文书
redis 限制内存使用大小的实现
2021/05/08 Redis
LeetCode189轮转数组python示例
2022/08/05 Python