[0-size Tensor No.67、304、330、251] Add 0-size Tensor support for flip (#72864)

co63oc · web-flow · commit 1fd0aa471267 · 2025-05-29T15:54:23.000+08:00
diff --git a/test/legacy_test/test_flip.py b/test/legacy_test/test_flip.py
@@ -157,6 +157,12 @@ def init_test_case(self):
         self.axis = [-1]
 
 
+class TestFlipOp_ZeroSize(TestFlipOp):
+    def init_test_case(self):
+        self.in_shape = (4, 0, 6, 3)
+        self.axis = [0, 2]
+
+
 # ----------------flip_fp16----------------
 def create_test_fp16_class(parent):
     @unittest.skipIf(
diff --git a/test/legacy_test/test_rot90_op.py b/test/legacy_test/test_rot90_op.py
@@ -315,5 +315,37 @@ def test_dygraph(self):
             )
 
 
+def create_test_zero_size_class(op_type, dtype, shape, axis):
+    class Cls(unittest.TestCase):
+        def test_zero_size(self):
+            paddle.disable_static()
+            numpy_tensor_1 = np.random.rand(*shape).astype(dtype)
+            paddle_x = paddle.to_tensor(numpy_tensor_1)
+            paddle_x.stop_gradient = False
+
+            paddle_api = eval(f"paddle.{op_type}")
+            paddle_out = paddle_api(paddle_x, axes=axis)
+            numpy_api = eval(f"np.{op_type}")
+            numpy_out = numpy_api(numpy_tensor_1, axes=axis)
+
+            np.testing.assert_allclose(
+                paddle_out.numpy(),
+                numpy_out,
+                1e-2,
+                1e-2,
+            )
+            np.testing.assert_allclose(
+                paddle_out.shape,
+                numpy_out.shape,
+            )
+
+    cls_name = f"{op_type}{dtype}_0SizeTest"
+    Cls.__name__ = cls_name
+    globals()[cls_name] = Cls
+
+
+create_test_zero_size_class("rot90", "float32", [3, 4, 0], (0, 1))
+create_test_zero_size_class("rot90", "int32", [3, 4, 0, 3, 4], (0, 1))
+
 if __name__ == "__main__":
     unittest.main()
