From 2255acf8e40685ccc82eb5094f9989d6e9f67195 Mon Sep 17 00:00:00 2001
From: liqitong-a <71805392+liqitong-a@users.noreply.github.com>
Date: Thu, 5 May 2022 15:49:01 +0800
Subject: [PATCH] Update test_corr.py

---
 python/paddle/fluid/tests/unittests/test_corr.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/test_corr.py b/python/paddle/fluid/tests/unittests/test_corr.py
index d0cb8507ffa77..99fd21c047b07 100644
--- 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():