pytorch: 给tensor删除或者添加维度为1的维度（squeeze和unsqueeze）

删除tensor中维度为1的维度

在pytorch中，用torch.squeeze()函数或者tensor的自身成员函数squeeze()去除维度为1的维度。

import torch
x = torch.randn(3,1,4,1,2)
a = x.squeeze(dim=1)  # 成员函数删除第二维度
print(a.shape)
b = torch.squeeze(x, dim=1)  # torch自带函数删除第二维度
print(b.shape)
c = torch.squeeze(x, dim=3)  # 删除第三维度
print(c.shape)
d = torch.squeeze(x)  # 若不标注删除第几维度，则会删除所有为1的维度
print(d.shape)

输出：
torch.Size([3, 4, 1, 2])
torch.Size([3, 4, 1, 2])
torch.Size([3, 1, 4, 2])
torch.Size([3, 4, 2])

注意：squeeze()函数只能删除维度为1的维度，若不为1，该操作无效，但不会报错

b = torch.unsqueeze(a, dim=1)
print(b.shape)

输出为：
torch.Size([3, 1, 4, 1, 2])

给tensor添加维度为1的维度

在pytorch中，用自带的torch.unsqueeze()和tensor的成员函数unsqueeze()可以为tensor添加维度为1的维度

e = torch.unsqueeze(x, dim=0)  # 在第一维度添加维度
print(e.shape)
f = x.unsqueeze(dim=0)
print(f.shape)

输出：
torch.Size([1, 3, 1, 4, 1, 2])
torch.Size([1, 3, 1, 4, 1, 2])

pytorch: 给tensor删除或者添加维度为1的维度（squeeze和unsqueeze）相关推荐

Pytorch中tensor维度和torch.max()函数中dim参数的理解
Pytorch中tensor维度和torch.max()函数中dim参数的理解维度参考了 https://blog.csdn.net/qq_41375609/article/details/106 ...
【PyTorch】 tensor.squeeze压缩维度
[PyTorch] tensor.squeeze压缩维度 1.背景 2.squeeze方法 3.示例 1.背景 pytorch的tensor维度扩展.压缩是一个常见的操作, tensor维度扩展参见如 ...
pyTorch中tensor运算
文章目录 PyTorch的简介 PyTorch中主要的包 PyTorch的安装使用GPU的原因使数据在GPU上运行什么使Tensor(张量) 一些术语介绍 Tensor的属性介绍(Rank,ax ...
【Pytorch】tensor类型数据.squeeze()和.unsqueeze()函数的简明教程（一看就会）
文章目录 1 squeeze 1.1 1.2 1.3 执行操作后需要写回 2 unsqueeze pytorch系列代码中常见的两个函数squeeze()和unsqueeze() 1 squeeze ...
pytorch方法,Tensor及其基本操作_重点
由于之前的草稿都没了,现在只有重写-. 我好痛苦本章只是对pytorch的常规操作进行一个总结,大家看过有脑子里有印象就好,知道有这么个东西,需要的时候可以再去详细的看,另外也还是需要在实战中多运用 ...
pytorch中tensor.mul()和mm()和matmul()
tensor.mul tensor.mul和tensor * tensor 都是将矩阵的对应位置的元素相乘,因此要求维度相同,点乘 torch.mul(input, other, *, out=Non ...
Pytorch List Tensor转Tensor,,reshape拼接等操作
Pytorch List Tensor转Tensor,reshape拼接等操作持续更新一些常用的Tensor操作,比如List,Numpy,Tensor之间的转换,Tensor的拼接,维度的变换等操 ...
动手学pytorch之tensor数据(一)
tensor数据类型 Tensor在使用时可以有不同的数据类型,官方给出了 7种CPU Tensor类型与8种GPU Tensor类型.16位半精度浮点是专为GPU模型设计的,以尽可能地节省GPU显存 ...
Pytorch中tensor.view().permute().contiguous()函数理解
Pytorch中tensor.view().permute().contiguous()函数理解 yolov3中有一行这样的代码,在此记录一下三个函数的含义 # 例子中batch_size为整型,le ...

pytorch: 给tensor删除或者添加维度为1的维度（squeeze和unsqueeze）

删除tensor中维度为1的维度

给tensor添加维度为1的维度

pytorch: 给tensor删除或者添加维度为1的维度（squeeze和unsqueeze）相关推荐

最新文章

热门文章