[ 参数不一致 ] torch.swapaxes

torch.swapaxes

torch.swapaxes(input, axis0, axis1)

paddle.transpose

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

其中 Pytorch 的 axis0、axis1 与 Paddle 用法不一致,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 输入 Tensor。 | | axis0 | - | Pytorch 转置的第一个维度,Paddle 无此参数,需要转写。 | | axis1 | - | Pytorch 转置的第二个维度,Paddle 无此参数,需要转写。 | | - | perm | Paddle 可通过 perm 参数,等价的实现 torch 的 axis0、axis1 的功能。|

转写示例

axis0、axis1 参数: 转置的维度设置

# PyTorch 写法:
torch.swapaxes(x, axis0=0, axis1=1)

# Paddle 写法:
paddle.transpose(x, perm=[1, 0, 2])

# 注:x 为 3D Tensor