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

torch.all ¶

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

         

          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'))