fixed bugs in gather test case (#2997)

PaddlePaddle · Nov 26, 2024 · 7738bd0 · 7738bd0
1 parent 4e8aced
commit 7738bd0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/gather_0_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/gather_0_func.py
@@ -33,14 +33,14 @@ def create_tensor_inputs():
     """
     paddle tensor
     """
-    inputs = (paddle.to_tensor(-1 + (1 - -1) * np.random.random([2, 3, 4, 4]).astype('float32'), dtype='float32', stop_gradient=False), )
+    inputs = (paddle.to_tensor(-1 + (1 - -1) * np.random.random([3, 3, 4, 4]).astype('float32'), dtype='float32', stop_gradient=False), )
     return inputs
 
 
 def create_numpy_inputs():
     """
     numpy array
     """
-    inputs = (-1 + (1 - -1) * np.random.random([2, 3, 4, 4]).astype('float32'), )
+    inputs = (-1 + (1 - -1) * np.random.random([3, 3, 4, 4]).astype('float32'), )
     return inputs
 
diff --git a/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/gather_base_func.py b/framework/e2e/PaddleLT_new/layerApicase/math_sublayer/gather_base_func.py
@@ -33,14 +33,14 @@ def create_tensor_inputs():
     """
     paddle tensor
     """
-    inputs = (paddle.to_tensor(-1 + (1 - -1) * np.random.random([2, 3, 4, 4]).astype('float32'), dtype='float32', stop_gradient=False), )
+    inputs = (paddle.to_tensor(-1 + (1 - -1) * np.random.random([3, 3, 4, 4]).astype('float32'), dtype='float32', stop_gradient=False), )
     return inputs
 
 
 def create_numpy_inputs():
     """
     numpy array
     """
-    inputs = (-1 + (1 - -1) * np.random.random([2, 3, 4, 4]).astype('float32'), )
+    inputs = (-1 + (1 - -1) * np.random.random([3, 3, 4, 4]).astype('float32'), )
     return inputs