[ 返回参数类型不一致 ]torch.equal

torch.equal

torch.equal(input,
            other)

paddle.equal_all

paddle.equal_all(x,
                 y,
                 name=None)

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

参数映射

PyTorch PaddlePaddle 备注
input x 表示输入的 Tensor ,仅参数名不一致。
other y 表示输入的 Tensor ,仅参数名不一致。

注:PyTorch 返回 bool 类型,Paddle 返回 0-D bool Tensor

转写示例

返回值

# PyTorch 写法
out = torch.equal(x, y)

# Paddle 写法
out = paddle.equal_all(x, y)
out = out.item()