pytorch transpose与permute函数

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]]])

你可能感兴趣的:(【python】,pytorch)