Having trouble understanding how torch.transpose is behaving

Nvm the problem was with my IDE