torch.unsqueeze()是一个压缩维度的方法，这个方法接受负参数有什么意义？

yellingf 2019-08-31 12:31:11

如何理解这个方法能够接受负参数？
输入的数据为X，输入的维度为dim
调用方法：torch.unsqueeze(X , dim = dim)
如果输入的参数dim为负，该参数会被转换成 dim + X.dim() + 1
该转换有什么依据？知道参数为负会被转换，为什么还要允许算法接收负参数？

...全文

224 回复打赏收藏转发到动态举报

写回复

回复

切换为时间正序

请发表友善的回复…

发表回复

Numpy 随机矩阵： np.random.randn(d0, d1, d2, …) 矩阵大小与形状： np.ndarray.size 与 np.dnarray.shape Pytorch 随机矩阵： torch.randn(d0, d1, d2, …) 添加维度： tensor.unsqueeze(0) 压缩维度: tensor.squeeze(0) 按维度拼接tensor： torch.cat(inputs, dim=0, …) 维度堆叠： torch.stack(inputs, dim=0) 张量排序索引： tensor.sort(descending=True) 返回一个tensor为

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 第一种方式，输入数据后直接加u

squeeze的用法主要就是对数据的维度进行压缩或者解压。先看torch.squeeze() 这个函数主要对数据的维度进行压缩，去掉维数为1的的维度，比如是一行或者一列这种，一个一行三列（1,3）的数去掉第一个维数为一的维度之后就变成（3）行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=torch.squeeze(a，N) a中去掉指定的定的维数为一的维度。再看torch.unsqueeze()这个函数主要是对数据维度进行扩充。给指定位置加上维数为一的维度，比如原本有个三行的数据（3

这三个函数的维度指定例如，某个数组是[2,4]的也就是2行4列的一个数组，将[2,4]作为一个列表来看，2对应得索引值是0或-2，同样4对应得索引值是1或-1 torch.squeeze()，torch.unsqueeze() torch.squeeze()是压缩维度值为1的维度，相反torch.unsqueeze()是新增加一个维度且设为1 squeeze() z=torch.randint(1,10,(1,4)) print(z.size()) print(z) y0=torch.squeeze(z,

squeeze的用法主要就是对数据的维度进行压缩或者解压。先看torch.squeeze() 这个函数主要对数据的维度进行压缩，去掉维数为1的的维度，比如是一行或者一列这种，一个一行三列（1,3）的数去掉第一个维数为一的维度之后就变成（3）行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=to...

37,720

社区成员

34,239

社区内容

发帖

与我相关

我的任务

社区管理员

加入社区

近7日
近30日
至今

加载中

查看更多榜单

试试用AI创作助手写篇文章吧

+ 用AI写文章