[API compatibility] torch.Tensor.prod torch.Tensor.reshape (#74559)

* [API compatibility] torch.Tensor.prod torch.Tensor.reshape

* fix reshape timeout

* fix reshape timeout

* fix reshape

Author: DongBaiYue

python/paddle/tensor/manipulation.py

@@ -26,8 +26,8 @@
 from paddle.tensor import fill_constant
 from paddle.utils.decorator_utils import (
     ParamAliasDecorator,
-    param_one_alias,
     param_two_alias,
+    reshape_decorator,
     view_decorator,
 )
 from paddle.utils.inplace_utils import inplace_apis_in_dygraph_only
@@ -4989,7 +4989,7 @@ def get_attr_expand_shape(list_expand_shape):
         return out
 
 
-@param_one_alias(["x", "input"])
+@reshape_decorator()
 def reshape(x: Tensor, shape: ShapeLike, name: str | None = None) -> Tensor:
     """
     Changes the shape of ``x`` without changing its data.

python/paddle/utils/decorator_utils.py

@@ -244,10 +244,8 @@ def process(
 """
     Usage Example:
     paddle.view(x=tensor_x, shape_or_dtype=[-1, 1, 3], name=None)
-
     tensor_x.view(paddle.float32) -> paddle.view(tensor_x, paddle.float32)
     tensor_x.view(dtype=paddle.float32) -> paddle.view(tensor_x, dtype=paddle.float32)
-
     tensor_x.view([-1, 1, 3]) -> paddle.view(tensor_x, [-1, 1, 3])
     tensor_x.view(-1, 1, 3) -> paddle.view(tensor_x, -1, 1, 3)
     tensor_x.view(size=[-1, 1, 3]) -> paddle.view(tensor_x, size=[-1, 1, 3])
@@ -273,3 +271,30 @@ def wrapper(*args, **kwargs):
         return wrapper
 
     return decorator
+
+
+def reshape_decorator():
+    """
+    Usage Example:
+    paddle.reshape(x=tensor_x, shape=[-1, 1, 3], name=None)
+    paddle.reshape(input=tensor_x, shape=[-1, 1, 3], name=None)
+    tensor_x.reshape([-1, 1, 3]) -> paddle.reshape(tensor_x, [-1, 1, 3])
+    tensor_x.reshape(-1, 1, 3) -> paddle.reshape(tensor_x, -1, 1, 3])
+    """
+
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            if ("input" in kwargs) and ("x" not in kwargs):
+                kwargs["x"] = kwargs.pop("input")
+            elif len(args) >= 2 and type(args[1]) is int:
+                if all(type(arg) is int for arg in args[1:]):
+                    kwargs["x"] = args[0]
+                    kwargs['shape'] = list(args[1:])
+                    args = ()
+            return func(*args, **kwargs)
+
+        wrapper.__signature__ = inspect.signature(func)
+        return wrapper
+
+    return decorator

test/legacy_test/test_prod_op.py

@@ -423,6 +423,35 @@ def test_gpu(self):
         with static_guard():
             self.run_static()
 
+    def test_tensor_prod(self):
+        """x.prod(axis=1) is equivalent to x.prod(dim=1)"""
+        axis_cases = [0, 1, -1]
+
+        def run_test_cases(place):
+            """Helper function to run test cases on specified device."""
+            for param_alias in ["axis", "dim"]:
+                for axis in axis_cases:
+                    input_tensor = paddle.to_tensor(self.input, place=place)
+                    kwargs = {param_alias: axis}
+
+                    result = input_tensor.prod(**kwargs)
+                    expected = np.prod(self.input, axis=axis)
+                    np.testing.assert_allclose(
+                        (
+                            result.numpy()
+                            if place.is_cpu_place()
+                            else result.cpu().numpy()
+                        ),
+                        expected,
+                        rtol=1e-05,
+                    )
+
+        with dygraph_guard():
+            run_test_cases(paddle.CPUPlace())
+
+            if paddle.base.core.is_compiled_with_cuda():
+                run_test_cases(paddle.CUDAPlace(0))
+
 
 if __name__ == "__main__":
     unittest.main()

test/legacy_test/test_reshape_op.py

@@ -915,6 +915,31 @@ def test_imperative(self):
         np.testing.assert_array_equal(out_2.numpy(), input.reshape([5, 10]))
         np.testing.assert_array_equal(out_3.numpy(), input.reshape(shape))
 
+    def test_tensor_reshape(self):
+        """The `shape` parameter accepts either variable arguments or a list/tuple.
+        For example, x.reshape(2, 5, 5) is equivalent to x.reshape([2, 5, 5]).
+        """
+
+        def run_test_cases(place):
+            """Helper function to run test cases on specified device."""
+            input = np.random.random([2, 25]).astype("float32")
+            input_tensor = paddle.to_tensor(input, place=place)
+
+            out_1 = input_tensor.reshape([2, 5, 5])
+            out_2 = input_tensor.reshape(2, 5, 5)
+
+            np.testing.assert_array_equal(
+                out_1.numpy(), input.reshape([2, 5, 5])
+            )
+            np.testing.assert_array_equal(
+                out_2.numpy(), input.reshape([2, 5, 5])
+            )
+
+        with base.dygraph.guard():
+            run_test_cases(paddle.CPUPlace())
+            if paddle.base.core.is_compiled_with_cuda():
+                run_test_cases(paddle.CUDAPlace(0))
+
 
 if __name__ == "__main__":
     paddle.enable_static()