>>> import torch
>>> a=torch.randn(1,3)
>>> print(a)
tensor([[-0.0837, -0.0136, -0.1506]])
>>> print(a.shape)
torch.Size([1, 3])
>>> b=torch.unsqueeze(a,2)
>>> print(b)
tensor([[[-0.0837],
[-0.0136],
[-0.1506]]])
>>> print(b.shape)
torch.Size([1, 3, 1])
>>> c=b.squeeze(0)
>>> print(c)
tensor([[-0.0837],
[-0.0136],
[-0.1506]])
>>> print(c.shape)
torch.Size([3, 1])