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

torch.all

torch.all(input, dim=None, keepdim=False, *, out=None)

paddle.all

paddle.all(x,
           axis=None,
           keepdim=False,
           name=None)

其中 Paddle 与 PyTorch 的 input 参数所支持的类型不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
input x 输入的多维 Tensor ,PyTorch 支持布尔和数值类型的输入,Paddle 仅支持布尔类型,需要转写。
dim axis 表示运算的维度,仅参数名不一致。
keepdim keepdim 是否在输出 Tensor 中保留减小的维度,参数完全一致。

转写示例

# PyTorch 写法
y = torch.all(x)

# Paddle 写法
y = paddle.all(x.astype('bool'))