Update test_corr.py

PaddlePaddle · May 5, 2022 · 2255acf · 2255acf
1 parent 51da5d6
commit 2255acf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_corr.py b/python/paddle/fluid/tests/unittests/test_corr.py
@@ -26,7 +26,7 @@ def numpy_corr(np_arr, rowvar=True, dtype='float64'):
 
 class Corr_Test(unittest.TestCase):
     def setUp(self):
-        self.shape = [20, 10]
+        self.shape = [4, 5]
 
     def test_tensor_corr_default(self):
         typelist = ['float64', 'float32']
@@ -84,13 +84,13 @@ def setUp(self):
 
 class Corr_Test3(Corr_Test):
     def setUp(self):
-        self.shape = [20, 10]
+        self.shape = [4, 5]
 
 
 # Input(x) only support N-D (1<=N<=2) tensor
 class Corr_Test4(unittest.TestCase):
     def setUp(self):
-        self.shape = [2, 5, 10]
+        self.shape = [2, 5, 2]
 
     def test_errors(self):
         def test_err():