diff options
Diffstat (limited to 'numpy/_array_api/searching_functions.py')
| -rw-r--r-- | numpy/_array_api/searching_functions.py | 17 |
1 files changed, 17 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..c4b6c58b5 --- /dev/null +++ b/numpy/_array_api/searching_functions.py @@ -0,0 +1,17 @@ +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) + +__all__ = ['argmax', 'argmin', 'nonzero', 'where'] |
