# jax.nn.softmax¶

jax.nn.softmax(x, axis=- 1)[source]

Softmax function.

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

$\mathrm{softmax}(x) = \frac{\exp(x_i)}{\sum_j \exp(x_j)}$
Parameters
Return type

Any