torch.transpose

torch.transpose
作用：一个tensor的两维度进行交换，只能有两个相互交换的位置参数

>>> x = torch.randn((2, 3,4))
>>> x.size()
torch.Size([2, 3, 4])

>>> y = torch.transpose(x,1,2) # 交换第二维和第三维
>>> y.size()
torch.Size([2, 4, 3])

>>> y = torch.transpose(x,0,1) # 交换第一维和第二维
>>> y.size()
torch.Size([3, 2, 4])