[ 参数不一致 ] torch.unflatten

torch.unflatten

torch.unflatten(input, dim, sizes)

paddle.reshape

paddle.reshape(x, shape, name=None)

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

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | input | x | 输入 Tensor。 | | dim | - | Pytorch 需要变换的维度,Paddle 无此参数,需要转写。 Paddle 可修改 shape 实现。 | | sizes | - | Pytorch 维度变换的新形状,Paddle 无此参数,需要转写。Paddle 可修改 shape 实现。 | | - | shape | Paddle 可通过 shape 参数,等价的实现 torch 的 dim、sizes 的功能。|

转写示例

dim、sizes 参数: 转置的维度设置

# PyTorch 写法:
torch.unflatten(input, -1, (2, 2))

# Paddle 写法:
paddle.reshape(input, shape=[3, 2, 2])

# 注:input 的形状为[3, 4]