Array API compatible wrapper for :py:func:`np.argmin `. See its docstring for more information.
(x: Array, /, *, axis: Optional[int] = None, keepdims: bool = False)
| 20 | |
| 21 | |
| 22 | def argmin(x: Array, /, *, axis: Optional[int] = None, keepdims: bool = False) -> Array: |
| 23 | """ |
| 24 | Array API compatible wrapper for :py:func:`np.argmin <numpy.argmin>`. |
| 25 | |
| 26 | See its docstring for more information. |
| 27 | """ |
| 28 | if x.dtype not in _real_numeric_dtypes: |
| 29 | raise TypeError("Only real numeric dtypes are allowed in argmin") |
| 30 | return Array._new(np.asarray(np.argmin(x._array, axis=axis, keepdims=keepdims))) |
| 31 | |
| 32 | |
| 33 | def nonzero(x: Array, /) -> Tuple[Array, ...]: |