# brainpy.math.softmax#

brainpy.math.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:
• x (ArrayType) – The input array.

• axis (int) – The axis or axes along which the softmax should be computed. The softmax output summed across these dimensions should sum to $$1$$. Either an integer or a tuple of integers.