[ 仅参数名不一致 ]torch.permute

torch.permute

torch.permute(input,
              dims)

paddle.transpose

paddle.transpose(x,
                 perm,
                 name=None)

两者功能一致且参数用法一致,仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 输入的多维 Tensor,仅参数名不一致。
dims perm dims 长度必须和 input 的维度相同,并依照 dims 中数据进行重排,仅参数名不一致。