1 files changed, 31 insertions, 0 deletions
diff --git a/numpy/array_api/_set_functions.py b/numpy/array_api/_set_functions.py
new file mode 100644
index 000000000..357f238f5
--- /dev/null
+++ b/numpy/array_api/_set_functions.py
@@ -0,0 +1,31 @@
+from __future__ import annotations
+
+from ._array_object import Array
+
+from typing import Tuple, Union
+
+import numpy as np
+
+
+def unique(
+    x: Array,
+    /,
+    *,
+    return_counts: bool = False,
+    return_index: bool = False,
+    return_inverse: bool = False,
+) -> Union[Array, Tuple[Array, ...]]:
+    """
+    Array API compatible wrapper for :py:func:`np.unique <numpy.unique>`.
+
+    See its docstring for more information.
+    """
+    res = np.unique(
+        x._array,
+        return_counts=return_counts,
+        return_index=return_index,
+        return_inverse=return_inverse,
+    )
+    if isinstance(res, tuple):
+        return tuple(Array._new(i) for i in res)
+    return Array._new(res)