pytorch:交换tensor的维度

在pytorch中， tensor有两个成员函数可以实现维度交换，分别时transpose()和permute()

transpose():

该函数可以交换tensor的任意两个维度，但是该函数一次只有两个参数，即一次只能交换两个维度。

import torch
x = torch.randn(8, 3, 5, 4)
y = x.transpose(1,2)  # 交换第二与第三维度
print(y.shape)

输出：
torch.Size([8, 5, 3, 4])

permute():

该函数可以随意交换任意维度，并且可以重新排列整合维度

z1 = x.permute(0,2,1,3)  # 交换第二与第三维度
print(z1.shape)
z2 = x.permute(3,0,2,1)  # 对原维度重新排列整合
print(z2.shape)

输出：
torch.Size([8, 5, 3, 4])
torch.Size([4, 8, 5, 3])