cupy.take¶

cupy.take(a, indices, axis=None, out=None)[source]¶

Takes elements of an array at specified indices along an axis.

This is an implementation of “fancy indexing” at single axis.

This function does not support mode option.

Parameters:	a (cupy.ndarray) – Array to extract elements. indices (int or array-like) – Indices of elements that this function takes. axis (int) – The axis along which to select indices. The flattened input is used by default. out (cupy.ndarray) – Output array. If provided, it should be of appropriate shape and dtype.
Returns:	The result of fancy indexing.
Return type:	cupy.ndarray