pytorch transpose与permute函数
pemute可以对高维,2阶或以上矩阵进行重排列,或者说转置,而transpose只能对两个维度进行调换
b = torch.tensor([[[1,4],[2,5]],[[3,7],[4,6]]])
b.shape
Out[36]: torch.Size([2, 2, 2])
b
Out[37]:
tensor([[[1, 4],
[2, 5]],
[[3, 7],
[4, 6]]])
b.permute(0,2,1)
Out[38]:
tensor([[[1, 2],
[4, 5]],
[[3, 4],
[7, 6]]])
b.transpose(2,1)
Out[40]:
tensor([[[1, 2],
[4, 5]],
[[3, 4],
[7, 6]]])