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: See also