[ 组合替代实现 ]torch.testing.assert_allclose¶

torch.testing.assert_allclose ¶

          torch.testing.assert_allclose(actual, expected, rtol=None, atol=None, equal_nan=True, msg='')

         

Paddle 无此 API，需要组合实现。

转写示例¶

          # PyTorch 写法
torch.testing.assert_allclose(actual, expected, rtol=rtol, atol=atol, equal_nan=True, msg='error messege')

# Paddle 写法
assert paddle.allclose(actual, expected, rtol=rtol, atol=atol, equal_nan=True).item(), 'error messege'