How does dim dim in squeeze and unsqueze work?
x = torch.zeros(2, 1, 2, 1, 2)
x.squeeze()
torch.Size([2, 2, 2])
y=torch.squeeze(x,1)
torch.Size([1, 2, 1, 2, 1, 2])   .//no change
y=torch.squeeze(x,2)
results :-
torch.Size([1, 2, 2, 1, 2])