diff --git a/python/paddle/fluid/tests/unittests/test_initializer.py b/python/paddle/fluid/tests/unittests/test_initializer.py
index 92135b113a0f995c3a4fa6da388b92610fa26907..952265e1195f5d3138143a0933d494c5d97ff466 100644
--- 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)