Searching Functions

Array API specification for functions for searching arrays.

A conforming implementation of the array API standard must provide and support the following functions.

Objects in API

argmax(x, /, *, axis=None, keepdims=False)

Returns the indices of the maximum values along a specified axis.

argmin(x, /, *, axis=None, keepdims=False)

Returns the indices of the minimum values along a specified axis.

nonzero(x, /)

Returns the indices of the array elements which are non-zero.

where(condition, x1, x2, /)

Returns elements chosen from x1 or x2 depending on condition.