.. role:: hidden
    :class: hidden-section
.. currentmodule:: brainstate.nn


softmax
=======

.. autoclass:: softmax
    :members: