jax.nn.log_softmax#
- jax.nn.log_softmax(x, axis=-1, where=None, initial=_UNSPECIFIED)[source]#
Log-Softmax function.
Computes the logarithm of the
softmax
function, which rescales elements to the range \([-\infty, 0)\).\[\mathrm{log\_softmax}(x)_i = \log \left( \frac{\exp(x_i)}{\sum_j \exp(x_j)} \right)\]- Parameters:
- Returns:
An array.
- Return type:
See also