[ 输入参数用法不一致 ] torch.transpose¶

torch.transpose ¶

          torch.transpose(input,
                dim0,
                dim1)

         

          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.transpose(x, dim0=0, dim1=1)

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

# 注：x 为 3D Tensor