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

torch.broadcast_shapes ¶

          torch.broadcast_shapes(*shapes)

         

          paddle.broadcast_shape(x_shape, y_shape)

         

其中功能一致, 参数用法不一致，具体如下：

PyTorch	PaddlePaddle	备注
*shapes	x_shape, y_shape	输入 Tensor 的 shape，输入参数数量不一致，PyTorch 可以输入多项，Paddle 输入为两项，需要多次调用进行转写。

           # PyTorch 写法:
torch.broadcast_shapes(x1, x2, x3)

# Paddle 写法:
paddle.broadcast_shape(paddle.broadcast_shape(x1, x2), x3)