2021-09-12

pytorch中tensor的squeeze()和unsqueeze()

squeeze()：
squeeze(arg)表示第arg维的维度值为1，则去掉该维度,否则tensor不变。

c = torch.randn(5,1)
c
tensor([[ 0.7312],
        [ 1.5407],
        [ 0.4843],
        [ 0.0443],
        [-0.8765]])

c.shape
结果：torch.Size([5, 1])

c.squeeze()
结果：tensor([ 0.7312,  1.5407,  0.4843,  0.0443, -0.8765])

unsqueeze()：

unsqueeze(arg)与squeeze(arg)作用相反，表示在第arg维增加一个维度值为1的维度。

在这里插入代码片