[ 输入参数类型不一致 ]torch.broadcast_tensors

torch.broadcast_tensors

torch.broadcast_tensors(*tensors)

paddle.broadcast_tensors

paddle.broadcast_tensors(inputs,
                         name=None)

两者功能一致但参数类型不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
*tensors inputs 一组输入 Tensor , PyTorch 参数 tensors 为可变参, Paddle 参数 inputs 为 list(Tensor) 或 tuple(Tensor) 的形式。

转写示例

*tensors: 一组输入 Tensor

# PyTorch 写法
torch.broadcast_tensors(x, y)

# Paddle 写法
paddle.broadcast_tensors([x, y])