[ 返回参数类型不一致 ]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()