chainerx.take¶
-
chainerx.
take
(a, indices, axis)¶ Takes elements from an array along an axis.
- Parameters
a (ndarray) – Source array.
indices (ndarray) – The indices of the values to extract. When indices are out of bounds, they are wrapped around.
axis (int) – The axis over which to select values.
mode (str) – Specifies how out-of-bounds indices will behave. ‘raise’ - raise an error ‘wrap’ - wrap around ‘clip’ - clip to the range
- Returns
Output array.
- Return type
Note
This function currently does not support
axis=None
Note
During backpropagation, this function propagates the gradient of the output array to the input array
a
.Note
The default mode for the native backend is ‘raise’, while for the cuda backend is ‘wrap’ in order to prevent device synchronization. ‘raise’ mode is currently not supported in the CUDA backend.
See also