Python——torch.unsqueeze

unsqueeze在指定位置处插入尺寸为1的维度。

注意：返回的张量与此张量共享相同的基础数据。

 

>>> x = torch.tensor([1, 2, 3, 4])
>>> torch.unsqueeze(x, 0)
tensor([[ 1,  2,  3,  4]])
>>> torch.unsqueeze(x, 1)
tensor([[ 1],
        [ 2],
        [ 3],
        [ 4]])

 

import torch

torch中的squeeze与unsqueeze作用是去除/添加维度为1的行

例如，a=torch.randn(2,3)

那么b=a.unsqueeze(0)，b为（1,2,3）的矩阵

类似的，b=a.unsqueeze(1)，b为（2,1,3）的矩阵

b=a.unsqueeze(2)，b为（2,3,1）的矩阵