cupy.take_along_axis#
- cupy.take_along_axis(a, indices, axis)[source]#
Take values from the input array by matching 1d index and data slices.
- Parameters:
a (cupy.ndarray) – Array to extract elements.
indices (cupy.ndarray) – Indices to take along each 1d slice of
a
.axis (int) – The axis to take 1d slices along.
- Returns:
The indexed result.
- Return type:
See also