x = torch.zeros(2, 1, 2, 1, 2) x.size() >>> torch.Size([2, 1, 2, 1, 2]) y = torch.squeeze(x) # remove 1 y.size() >>> torch.Size([2, 2, 2]) y = torch.squeeze(x, 0) y.size() >>> torch.Size([2, 1, 2, 1, 2]) y = torch.squeeze(x, 1) y.size() >>> torch.Size([2, 2, 1, 2])
ft = torch.Tensor([0, 1, 2]) print(ft.shape) >>> torch.Size([3]) print(ft.unsqueeze(0)) # 0 means first dimension print(ft.unsqueeze(0).shape) >>> tensor([[0., 1., 2.]]) >>> torch.Size([1, 3])