【Hackathon 9th Sprint No.1】add api compatibility for paddle.unique - part

python/paddle/tensor/manipulation.py

@@ -3770,13 +3770,15 @@ def unique(
 ) -> Tensor | tuple[Tensor, ...]: ...
 
 
+@param_two_alias(["x", "input"], ["axis", "dim"])
 def unique(
     x,
     return_index=False,
     return_inverse=False,
     return_counts=False,
     axis=None,
     dtype="int64",
+    sorted=True,
     name=None,
 ):
     r"""
@@ -3793,6 +3795,7 @@ def unique(
             Default: None.
         dtype(str|paddle.dtype|np.dtype, optional): The date type of `indices` or `inverse` tensor: int32 or int64.
             Default: int64.
+        sorted(bool, optional): Does not affect the return result, used for compatibility.
         name(str|None, optional): Name for the operation. For more information, please refer to
             :ref:`api_guide_Name`. Default: None.
 

test/legacy_test/test_unique.py

@@ -500,5 +500,50 @@ def test_dygraph_api_out(self):
         np.testing.assert_allclose(out.numpy(), expected_out)
 
 
+class TestUniqueAPI_Compatibility(unittest.TestCase):
+    def setUp(self):
+        self.x_np = np.random.random(size=[3, 5]).astype("float32")
+        self.place = (
+            core.CUDAPlace(0)
+            if core.is_compiled_with_cuda()
+            else core.CPUPlace()
+        )
+
+    def test_dygraph(self):
+        paddle.disable_static()
+        out = paddle.unique(paddle.to_tensor(self.x_np))
+        expected_out = np.unique(self.x_np)
+        np.testing.assert_allclose(out.numpy(), expected_out)
+
+    def test_static(self):
+        paddle.enable_static()
+        x = paddle.static.data(name='x', shape=[-1, 5], dtype='float32')
+        out1 = paddle.unique(x)
+        out2 = paddle.unique(input=x)
+        exe = paddle.static.Executor(self.place)
+        res = exe.run(
+            feed={
+                'x': self.x_np.reshape(3, 5),
+            },
+            fetch_list=[out1, out2],
+        )
+        expected_out = np.unique(self.x_np)
+        for result in res:
+            np.testing.assert_array_equal(result, expected_out)
+        paddle.disable_static()
+
+    def test_dygraph_sorted(self):
+        paddle.disable_static()
+        out = paddle.unique(paddle.to_tensor(self.x_np), sorted=True)
+        expected_out = np.unique(self.x_np)
+        np.testing.assert_allclose(out.numpy(), expected_out)
+
+    def test_dygraph_axis(self):
+        paddle.disable_static()
+        out = paddle.unique(paddle.to_tensor(self.x_np), sorted=True, dim=1)
+        expected_out = np.unique(self.x_np, axis=1)
+        np.testing.assert_allclose(out.numpy(), expected_out)
+
+
 if __name__ == "__main__":
     unittest.main()