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


Softmax2d
=========

.. autoclass:: Softmax2d
    :members: