tiny\_diff.modules.attention
============================

.. automodule:: tiny_diff.modules.attention

   
   
   

   
   
   

   
   
   .. rubric:: Classes

   .. autosummary::
      :toctree:
      :template: custom-class-template.rst
   
      AttnFlatten
      CrossVisualAttention
      SelfVisualAttention
      VisualAttention