diff options
Diffstat (limited to 'numpy/_array_api/_searching_functions.py')
| -rw-r--r-- | numpy/_array_api/_searching_functions.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/numpy/_array_api/_searching_functions.py b/numpy/_array_api/_searching_functions.py index 62763eaca..4eed66c48 100644 --- a/numpy/_array_api/_searching_functions.py +++ b/numpy/_array_api/_searching_functions.py @@ -1,13 +1,33 @@ import numpy as np def argmax(x, /, *, axis=None, keepdims=False): + """ + Array API compatible wrapper for :py:func:`np.argmax <numpy.argmax>`. + + See its docstring for more information. + """ return np.argmax(x, axis=axis, keepdims=keepdims) def argmin(x, /, *, axis=None, keepdims=False): + """ + Array API compatible wrapper for :py:func:`np.argmin <numpy.argmin>`. + + See its docstring for more information. + """ return np.argmin(x, axis=axis, keepdims=keepdims) def nonzero(x, /): + """ + Array API compatible wrapper for :py:func:`np.nonzero <numpy.nonzero>`. + + See its docstring for more information. + """ return np.nonzero(x) def where(condition, x1, x2, /): + """ + Array API compatible wrapper for :py:func:`np.where <numpy.where>`. + + See its docstring for more information. + """ return np.where(condition, x1, x2) |