Pytorch 扩展Tensor维度、压缩Tensor维度的方法
作者:抚琴尘世客 时间:2022-05-29 03:35:32
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)
结果显示
第二种方式,用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)
结果显示
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)
结果显示
第二种方式,用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)
结果显示
来源:https://www.cnblogs.com/haifwu/p/12829900.html
标签:Pytorch,Tensor,维度
0
投稿
猜你喜欢
python 实现将小图片放到另一个较大的白色或黑色背景图片中
2022-03-26 00:21:36
django 解决manage.py migrate无效的问题
2021-08-18 02:28:51
Python自动化测试中yaml文件读取操作
2021-11-19 11:55:43
从开发人员角度看IE8的开发新特性
2010-02-26 10:48:00
PHP 引用的概念
2023-11-14 21:24:28
基于Python实现二维图像双线性插值
2023-08-13 07:46:45
python模块简介之有序字典(OrderedDict)
2023-12-14 07:46:46
如何在PyCharm中配置使用Anaconda环境
2021-01-28 14:16:53
巧制可全屏拖动的图片
2008-05-09 19:34:00
网站开发防止中文乱码需要了解的codepage的重要性小结
2011-03-03 11:24:00
Python面向对象编程(一)
2023-04-27 15:02:13
exe反编译为.py文件的方法
2022-06-10 16:18:16
ASP 日期的加减运算实现代码
2011-03-08 10:47:00
javabean servlet jsp实现分页功能代码解析
2023-06-13 15:21:24
利用Pycharm + Django搭建一个简单Python Web项目的步骤
2021-10-21 15:35:47
详解Python 模拟实现生产者消费者模式的实例
2022-04-17 10:35:51
python连接sql server乱码的解决方法
2023-06-27 07:25:17
asp使用ServerVariables集合
2008-02-27 13:22:00
在python中利用GDAL对tif文件进行读写的方法
2022-03-25 08:08:07
使用golang编写一个并发工作队列
2023-09-02 20:54:08