jax-js
    Preparing search index...

    Function softmax

    • Softmax function. Computes the function which rescales elements to the range [0, 1] such that the elements along axis sum to 1.

      If axis is not specified, it defaults to the last axis.

      Reference: https://en.wikipedia.org/wiki/Softmax_function

      Parameters

      • x: ArrayLike
      • axis: null | number | number[] = -1

      Returns Array