Back to Numerical Arrays (NumPy)
np.take_along_axis
Take values from the input array by matching 1d index and data slices.
Syntax
np.take_along_axis(arr, indicesArr, axis)
Description
Take values from the input array by matching 1d index and data slices.
Parameters
| Name | Description |
|---|---|
| arr | - Source array |
| indicesArr | NDArray |
| axis | - The axis to take along |
Returns
Promise<NDArray>