jax-js
    Preparing search index...

    Function logSoftmax

    • Log-Softmax function.

      Computes the logarithm of the softmax function, which rescales elements to the range [-infinity, 0).

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

      Parameters

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

      Returns Array