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
Log-Softmax function.
Computes the logarithm of the softmax function, which rescales elements to the range \([-\infty, 0)\).
softmax
x (JaxArray, jnp.ndarray) – The input array.
axis (int) – The axis or axes along which the log_softmax should be computed. Either an integer or a tuple of integers.
log_softmax