emmi.modules.attention.utils
============================

.. py:module:: emmi.modules.attention.utils


Functions
---------

.. autoapisummary::

   emmi.modules.attention.utils.apply_init_method


Module Contents
---------------

.. py:function:: apply_init_method(module, proj_weight, init_method)

   Apply an initialization function to all applicable sub-modules of a given module.

   :param module: The nn.Module instance to initialize.
   :param init_fn: The initialization function to apply to each sub-module.