fix random failure (#27996)

78b1026f · Leo Chen · GitHub · ffcc1175 · 78b1026f
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_initializer.py python/paddle/fluid/tests/unittests/test_initializer.py +2 -2

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_initializer.py
+++ b/python/paddle/fluid/tests/unittests/test_initializer.py
@@ -599,9 +599,9 @@ class TestUniformInitializerDygraph(unittest.TestCase):
        """
        paddle.disable_static()

-        tensor = paddle.zeros([1024, 1024])
+        tensor = paddle.zeros([1024, 1024, 16])
        tensor.stop_gradient = False
-        self.assertTrue(np.allclose(np.zeros((1024, 1024)), tensor.numpy()))
+        self.assertTrue(np.allclose(np.zeros((1024, 1024, 16)), tensor.numpy()))

        uniform_ = paddle.nn.initializer.Uniform()
        uniform_(tensor)