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


assign_state_values
===================

.. autoclass:: assign_state_values
    :members: