numpy.array_api._searching_functions

module documentation
(source)

Undocumented

Function argmax Array API compatible wrapper for np.argmax.
Function argmin Array API compatible wrapper for np.argmin.
Function nonzero Array API compatible wrapper for np.nonzero.
Function where Array API compatible wrapper for np.where.
def argmax(x: Array, /, *, axis: Optional[int] = None, keepdims: bool = False) -> Array: (source)

Array API compatible wrapper for np.argmax.

See its docstring for more information.

def argmin(x: Array, /, *, axis: Optional[int] = None, keepdims: bool = False) -> Array: (source)

Array API compatible wrapper for np.argmin.

See its docstring for more information.

def nonzero(x: Array, /) -> Tuple[Array, ...]: (source)

Array API compatible wrapper for np.nonzero.

See its docstring for more information.

def where(condition: Array, x1: Array, x2: Array, /) -> Array: (source)

Array API compatible wrapper for np.where.

See its docstring for more information.