torch 的 unsqueeze用法

import torcha=torch.tensor([[3.0000, 3.0000],[3.0000, 4.0000],[3.6000, 3.0000],[3.5000, 3.0000]])

个人以为这个unsqueeze方法有点废话，完全可以用reshape 实现，鸡肋！


a.reshape([a.shape[0],1,a.shape[1]])==a.unsqueeze(1)a.reshape([1,a.shape[0],1,a.shape[1]])==a.unsqueeze(0)a.reshape([a.shape[0],a.shape[1] ,1])==a.unsqueeze(2)

a.reshape([a.shape[0],1,a.shape[1]])==a.unsqueeze(1)

'''0
m ,n------> 1,m,n1
m ,n------> m,1,n2
m ,n------> m,n,2k
m ,.....,n------> m,...,k,k+1,n
'''

print('a=',a)
print('a.shape=',a.shape)
print('-'*30+'分割线'+'-'*30)print('a.unsqueeze(0).shape=',a.unsqueeze(0).shape)print('a.unsqueeze(0)=',a.unsqueeze(0))print('-'*30+'分割线'+'-'*30)print('a.unsqueeze(1).shape=',a.unsqueeze(1).shape)print('a.unsqueeze(1)=',a.unsqueeze(1))print('-'*30+'分割线'+'-'*30)print('a.unsqueeze(2).shape=',a.unsqueeze(2).shape)print('a.unsqueeze(2)=',a.unsqueeze(2))

在这里是一个二位矩阵　可以在[0,1,2]三个维度上

a= tensor([[3.0000, 3.0000],[3.0000, 4.0000],[3.6000, 3.0000],[3.5000, 3.0000]])
a.shape= torch.Size([4, 2])
------------------------------分割线------------------------------
a.unsqueeze(0).shape= torch.Size([1, 4, 2])
a.unsqueeze(0)= tensor([[[3.0000, 3.0000],[3.0000, 4.0000],[3.6000, 3.0000],[3.5000, 3.0000]]])
------------------------------分割线------------------------------
a.unsqueeze(1).shape= torch.Size([4, 1, 2])
a.unsqueeze(1)= tensor([[[3.0000, 3.0000]],[[3.0000, 4.0000]],[[3.6000, 3.0000]],[[3.5000, 3.0000]]])
------------------------------分割线------------------------------
a.unsqueeze(2).shape= torch.Size([4, 2, 1])
a.unsqueeze(2)= tensor([[[3.0000],[3.0000]],[[3.0000],[4.0000]],[[3.6000],[3.0000]],[[3.5000],[3.0000]]])

一个是顺序，一个逆序，形成一个闭环

'''
-3 -2 -1
0   1  2
'''
print('a.unsqueeze(-3)==a.unsqueeze(0)',a.unsqueeze(-3)==a.unsqueeze(0))print('a.unsqueeze(-2)==a.unsqueeze(1)',a.unsqueeze(-2)==a.unsqueeze(1))print('a.unsqueeze(-1)==a.unsqueeze(2)',a.unsqueeze(-1)==a.unsqueeze(2))

a.unsqueeze(-3)==a.unsqueeze(0) tensor([[[True, True],[True, True],[True, True],[True, True]]])
a.unsqueeze(-2)==a.unsqueeze(1) tensor([[[True, True]],[[True, True]],[[True, True]],[[True, True]]])
a.unsqueeze(-1)==a.unsqueeze(2) tensor([[[True],[True]],[[True],[True]],[[True],[True]],[[True],[True]]])

torch 的 unsqueeze用法相关推荐

conv2d的输入_pytorch1.0中torch.nn.Conv2d用法详解
Conv2d的简单使用 torch 包 nn 中 Conv2d 的用法与 tensorflow 中类似,但不完全一样. 在 torch 中,Conv2d 有几个基本的参数,分别是 in_channel ...
torch.rand、torch.randn及torch.normal的用法
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档 torch.rand.torch.randn及torch.normal的用法一.torch.rand 二.torch.randn 三 ...
torch.sequeeze 和 torch.unsequeeze 的用法
torch.sequeeze 对维度进行压缩,以下为示例: import torcha = torch.zeros(2, 6, 1, 3, 1) # torch.Size([2, 6, 1, 3, 1 ...
pytorch中的torch.tensor.repeat以及torch.tensor.expand用法
文章目录 torch.tensor.expand torch.tensor.repeat torch.tensor.expand 先看招 import torch x = torch.tensor([ ...
torch.nn.Conv2d() 用法讲解
本文是深度学习框架 pytorch 的API : torch.nn.Conv2d() 函数的用法.本博客介绍了 torch.nn.Conv2d() 各个参数的含义和用法,学会使用 pytorch 创建 ...
torch.roll() 函数用法
Pytorch 官方文档:https://pytorch.org/docs/master/generated/torch.roll.htmlhttps://pytorch.org/docs/maste ...
torch.where()的用法以及例子
1.用法 torch.where()函数的作用是按照一定的规则合并两个tensor类型. torch.where(condition,a,b)其中输入参数condition:条件限制,如果满足条件, ...
pytorch torch.sort()的用法以及例子
1.作用对tensor中元素排序 2.用法 dim = -1,按照行排序,dim= 1按照列排序,descending=True,则递减排序,否则递增 3.例子按照行排序 logits = tor ...
PyTorch torch.Tensor.contiguous() 用法与理解
中文文档: contiguous() → Tensor 返回一个内存连续的有相同数据的 tensor,如果原 tensor 内存连续则返回原 tensor 英文文档: contiguo ...

torch 的 unsqueeze用法

个人以为这个unsqueeze方法有点废话，完全可以用reshape 实现，鸡肋！

在这里是一个二位矩阵　可以在[0,1,2]三个维度上

一个是顺序，一个逆序，形成一个闭环

torch 的 unsqueeze用法相关推荐

最新文章

热门文章

torch 的 unsqueeze用法

个人以为这个unsqueeze方法有点废话，完全可以用reshape 实现，鸡肋！

在这里是一个二位矩阵 可以在[0,1,2]三个维度上

一个是顺序，一个逆序，形成一个闭环

torch 的 unsqueeze用法相关推荐

最新文章

热门文章

在这里是一个二位矩阵　可以在[0,1,2]三个维度上