Add view_as_complex and view_as_real

Author: LittleHeroZZZX

python/paddle/__init__.py

@@ -363,6 +363,8 @@
     unstack,
     view,
     view_as,
+    as_complex as view_as_complex,
+    as_real as view_as_real,
     vsplit,
     vstack,
 )
@@ -1159,7 +1161,9 @@
     'acosh',
     'atanh',
     'as_complex',
+    'view_as_complex',
     'as_real',
+    'view_as_real',
     'diff',
     'angle',
     'fmax',

python/paddle/tensor/manipulation.py

@@ -6187,7 +6187,7 @@ def _var_to_list(var):
     out = x.matmul(y).reshape(shape_out)
     return out
 
-
+@ParamAliasDecorator({"x": ["input"]})
 def as_complex(x: Tensor, name: str | None = None) -> Tensor:
     """Transform a real tensor to a complex tensor.
 
@@ -6240,7 +6240,7 @@ def as_complex(x: Tensor, name: str | None = None) -> Tensor:
         )
         return out
 
-
+@ParamAliasDecorator({"x": ["input"]})
 def as_real(x: Tensor, name: str | None = None) -> Tensor:
     """Transform a complex tensor to a real tensor.
 

test/legacy_test/test_complex_view_op.py

@@ -33,7 +33,7 @@ def ref_view_as_real(x):
     return np.stack([x.real, x.imag], -1)
 
 
-class TestViewAsComplexOp(OpTest):
+class TestAsComplexOp(OpTest):
     def setUp(self):
         self.op_type = "as_complex"
         self.python_api = paddle.as_complex
@@ -53,7 +53,7 @@ def test_check_grad(self):
         )
 
 
-class TestViewAsRealOp(OpTest):
+class TestAsRealOp(OpTest):
     def setUp(self):
         self.op_type = "as_real"
         real = np.random.randn(10, 10).astype("float64")
@@ -75,7 +75,7 @@ def test_check_grad(self):
         )
 
 
-class TestViewAsComplexAPI(unittest.TestCase):
+class TestAsComplexAPI(unittest.TestCase):
     def setUp(self):
         self.x = np.random.randn(10, 10, 2)
         self.out = ref_view_as_complex(self.x)
@@ -98,7 +98,7 @@ def test_static(self):
         np.testing.assert_allclose(self.out, out_np, rtol=1e-05)
 
 
-class TestViewAsRealAPI(unittest.TestCase):
+class TestAsRealAPI(unittest.TestCase):
     def setUp(self):
         self.x = np.random.randn(10, 10) + 1j * np.random.randn(10, 10)
         self.out = ref_view_as_real(self.x)
@@ -121,7 +121,7 @@ def test_static(self):
         np.testing.assert_allclose(self.out, out_np, rtol=1e-05)
 
 
-class TestViewAsRealAPI_ZeroSize(unittest.TestCase):
+class TestAsRealAPI_ZeroSize(unittest.TestCase):
     def setUp(self):
         self.x = np.random.randn(10, 0) + 1j * np.random.randn(10, 0)
         self.out = ref_view_as_real(self.x)
@@ -137,5 +137,50 @@ def test_dygraph(self):
                 np.testing.assert_allclose(x_tensor.grad.shape, x_tensor.shape)
 
 
+class TestViewAsComplexAPI(unittest.TestCase):
+    def setUp(self):
+        self.x = np.random.randn(10, 10, 2)
+        self.out = ref_view_as_complex(self.x)
+
+    def test_dygraph(self):
+        with dygraph.guard():
+            x = paddle.to_tensor(self.x)
+            out = paddle.view_as_complex(x)
+            out_np = out.numpy()
+            self.assertEqual(out.data_ptr(), x.data_ptr())
+        np.testing.assert_allclose(self.out, out_np, rtol=1e-05)
+
+
+class TestViewAsRealAPI(unittest.TestCase):
+    def setUp(self):
+        self.x = np.random.randn(10, 10) + 1j * np.random.randn(10, 10)
+        self.out = ref_view_as_real(self.x)
+
+    def test_dygraph(self):
+        with dygraph.guard():
+            x = paddle.to_tensor(self.x)
+            out = paddle.view_as_real(x)
+            out_np = out.numpy()
+            self.assertEqual(out.data_ptr(), x.data_ptr())
+        np.testing.assert_allclose(self.out, out_np, rtol=1e-05)
+
+
+class TestViewAsRealAPI_ZeroSize(unittest.TestCase):
+    def setUp(self):
+        self.x = np.random.randn(10, 0) + 1j * np.random.randn(10, 0)
+        self.out = ref_view_as_real(self.x)
+
+    def test_dygraph(self):
+        for place in get_places():
+            with dygraph.guard(place):
+                x_tensor = paddle.to_tensor(self.x)
+                x_tensor.stop_gradient = False
+                out = paddle.view_as_real(x_tensor)
+                np.testing.assert_allclose(self.out, out.numpy(), rtol=1e-05)
+                self.assertEqual(out.data_ptr(), x_tensor.data_ptr())
+                out.sum().backward()
+                np.testing.assert_allclose(x_tensor.grad.shape, x_tensor.shape)
+
+
 if __name__ == "__main__":
     unittest.main()