[ 参数不一致 ]torch.Tensor.transpose¶

torch.Tensor.transpose ¶

          torch.Tensor.transpose(dim0, dim1)

         

paddle.Tensor.transpose ¶

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

         

Pytorch 的 dim0, dim1 与 Paddle 的 perm 用法不同，具体如下：

参数映射¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | dim0, dim1 | perm | torch 的 dim0 与 dim1 表示要交换的两个维度, 为整数。 paddle 的 perm 表示重排的维度序列，为 list/tuple 。需要转写。|

转写示例¶

dim0, dim1: 表示要交换的两个维度¶

           # pytorch
x = torch.randn(2, 3, 5)
y = x.transpose(0, 1)

# paddle
x = paddle.randn([2, 3, 5])
y = x.transpose(perm=[1, 0, 2])

          