[ 参数不⼀致 ] torch.swapdims¶

torch.swapdims¶

torch.swapdims(input,
                dim0,
                dim1)

paddle.transpose¶

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

其中 Pytorch 的 dim0、dim1 与 Paddle 用法不一致，具体如下：

参数映射¶

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

转写示例¶

# PyTorch 写法:
torch.swapdims(x, dim0=0, dim1=1)

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

# 注：x 为 3D Tensor