fix random failure (#27996)

4 years ago · 78b1026fa3
parent ffcc11758a
commit 78b1026fa3
1 changed files with 2 additions and 2 deletions
--- 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)