fix cholesky of test_math_op_patch_var_base (#27591)

c5b6e44b · Zhou Wei · GitHub · 9b7ebf10 · c5b6e44b
显示空白变更内容
内联并排

Showing with 5 addition and 3 deletion

python/paddle/fluid/tests/unittests/test_math_op_patch_var_base.py ...ddle/fluid/tests/unittests/test_math_op_patch_var_base.py +5 -3

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_math_op_patch_var_base.py
+++ b/python/paddle/fluid/tests/unittests/test_math_op_patch_var_base.py
@@ -341,10 +341,12 @@ class TestMathOpPatchesVarBase(unittest.TestCase):
            np.array_equal(x.rank().numpy(), paddle.rank(x).numpy()))
        self.assertTrue(
            np.array_equal(x[0].t().numpy(), paddle.t(x[0]).numpy()))
-        m = paddle.to_tensor(np.random.uniform(1, 2, [3, 3]), 'float32')
+        d = paddle.to_tensor([[1.2285208, 1.3491015, 1.4899898],
-        m = m.matmul(m.t())
+                              [1.30058, 1.0688717, 1.4928783],
+                              [1.0958099, 1.3724753, 1.8926544]])
+        d = d.matmul(d.t())
        self.assertTrue(
-            np.array_equal(m.cholesky().numpy(), paddle.cholesky(m).numpy()))
+            np.array_equal(d.cholesky().numpy(), paddle.cholesky(d).numpy()))
        self.assertTrue(
            np.array_equal(x.is_empty().numpy(), paddle.is_empty(x).numpy()))