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

torch.Tensor.reshape ¶

          torch.Tensor.reshape(*shape)

         

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

         

PyTorch 的 *shape 相比于 Paddle 的 shape 额外支持可变参数的用法，具体如下：

PyTorch	PaddlePaddle	备注
*shape	shape	torch 支持可变参数或 list/tuple，paddle 仅支持 list/tuple。对于可变参数的用法，需要转写。

           # pytorch
x = torch.randn(2, 3, 5)
y = x.reshape(6, 5)

# paddle
x = paddle.randn([2, 3, 5])
y = x.reshape((6, 5))

          