Indexing Functions

Array API specification for functions for indexing arrays.

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

Objects in API

take(x, indices, /, *, axis=None)

Returns elements of an array along an axis.

take_along_axis(x, indices, /, *, axis=-1)

Returns elements from an array at the one-dimensional indices specified by indices along a provided axis.