jax-js
    Preparing search index...

    Function takeAlongAxis

    • Take values from an array by matching 1D index slices along an axis.

      The input array and indices must have the same rank. Dimensions other than axis follow standard broadcasting rules, and the output shape is the broadcasted shape with the size of indices along axis.

      Parameters

      Returns Array