jax-js
    Preparing search index...

    Function argmax

    • Returns the indices of the maximum values along an axis.

      By default, index is into the flatted array, otherwise it is along the specified axis.

      Parameters

      • a: ArrayLike
      • Optionalaxis: number
      • Optionalopts: { keepdims?: boolean }

      Returns Array