diff --git a/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py b/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py
index 424239442f9e6773ba6d7099e8cc148f9b3d9b9e..0db669bc1f9508f90111d6c04e6730ef3433de29 100644
--- a/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py
+++ b/python/paddle/fluid/tests/unittests/test_bilinear_tensor_product_op.py
@@ -23,9 +23,9 @@ class TestBilinearTensorProductOp(OpTest):
     def setUp(self):
         self.op_type = "bilinear_tensor_product"
         batch_size = 6
-        size0 = 30
-        size1 = 20
-        size2 = 100
+        size0 = 5
+        size1 = 4
+        size2 = 5
         a = np.random.random((batch_size, size0)).astype("float64")
         b = np.random.random((batch_size, size1)).astype("float64")
         w = np.random.random((size2, size0, size1)).astype("float64")