Quickstart
BDP Tutorials
brainpy.math
Toolboxes
Advanced Tutorials
API Documentation
brainpy.base
brainpy.math.compat
brainpy.dyn
brainpy.nn
brainpy.analysis
brainpy.integrators
brainpy.datasets
brainpy.inputs
brainpy.connect
brainpy.initialize
brainpy.losses
brainpy.optimizers
brainpy.measure
brainpy.running
brainpy.tools
brainpy.compat
Softmax function.
Computes the function which rescales elements to the range \([0, 1]\) such that the elements along axis sum to \(1\).
axis
x (JaxArray, jnp.ndarray) – 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.