diff --git a/python/paddle/fluid/tests/unittests/test_corr.py b/python/paddle/fluid/tests/unittests/test_corr.py index 85c2801b97bab..25f17fc75d538 100644 --- a/python/paddle/fluid/tests/unittests/test_corr.py +++ b/python/paddle/fluid/tests/unittests/test_corr.py @@ -44,7 +44,7 @@ def test_tensor_corr_default(self): tensor = paddle.to_tensor(np_arr, place=p) corr = paddle.linalg.corrcoef(tensor) np_corr = numpy_corr(np_arr, rowvar=True, dtype=dtype) - self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-2)) + self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-4)) def test_tensor_corr_rowvar(self): typelist = ['float64', 'float32'] @@ -63,7 +63,7 @@ def test_tensor_corr_rowvar(self): tensor = paddle.to_tensor(np_arr, place=p) corr = paddle.linalg.corrcoef(tensor, rowvar=False) np_corr = numpy_corr(np_arr, rowvar=False, dtype=dtype) - self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-2)) + self.assertTrue(np.allclose(np_corr, corr.numpy(), atol=1.e-4)) # Input(x) only support N-D (1<=N<=2) tensor