summaryrefslogtreecommitdiff
path: root/numpy/_array_api/_searching_functions.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/_array_api/_searching_functions.py')
-rw-r--r--numpy/_array_api/_searching_functions.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/numpy/_array_api/_searching_functions.py b/numpy/_array_api/_searching_functions.py
new file mode 100644
index 000000000..c6035ca77
--- /dev/null
+++ b/numpy/_array_api/_searching_functions.py
@@ -0,0 +1,15 @@
+def argmax(x, /, *, axis=None, keepdims=False):
+ from .. import argmax
+ return argmax(x, axis=axis, keepdims=keepdims)
+
+def argmin(x, /, *, axis=None, keepdims=False):
+ from .. import argmin
+ return argmin(x, axis=axis, keepdims=keepdims)
+
+def nonzero(x, /):
+ from .. import nonzero
+ return nonzero(x)
+
+def where(condition, x1, x2, /):
+ from .. import where
+ return where(condition, x1, x2)