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

.. currentmodule:: tiny_diff.modules.attention

.. autoclass:: AttnFlatten
   :members:                                    <-- add at least this line
   :show-inheritance:                           <-- plus I want to show inheritance...
   :no-inherited-members:                          <-- ...and inherited members too

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~AttnFlatten.__init__
      ~AttnFlatten.add_module
      ~AttnFlatten.apply
      ~AttnFlatten.bfloat16
      ~AttnFlatten.buffers
      ~AttnFlatten.children
      ~AttnFlatten.compile
      ~AttnFlatten.cpu
      ~AttnFlatten.cuda
      ~AttnFlatten.double
      ~AttnFlatten.eval
      ~AttnFlatten.extra_repr
      ~AttnFlatten.float
      ~AttnFlatten.forward
      ~AttnFlatten.get_buffer
      ~AttnFlatten.get_extra_state
      ~AttnFlatten.get_parameter
      ~AttnFlatten.get_submodule
      ~AttnFlatten.half
      ~AttnFlatten.ipu
      ~AttnFlatten.load_state_dict
      ~AttnFlatten.modules
      ~AttnFlatten.named_buffers
      ~AttnFlatten.named_children
      ~AttnFlatten.named_modules
      ~AttnFlatten.named_parameters
      ~AttnFlatten.parameters
      ~AttnFlatten.register_backward_hook
      ~AttnFlatten.register_buffer
      ~AttnFlatten.register_forward_hook
      ~AttnFlatten.register_forward_pre_hook
      ~AttnFlatten.register_full_backward_hook
      ~AttnFlatten.register_full_backward_pre_hook
      ~AttnFlatten.register_load_state_dict_post_hook
      ~AttnFlatten.register_module
      ~AttnFlatten.register_parameter
      ~AttnFlatten.register_state_dict_pre_hook
      ~AttnFlatten.requires_grad_
      ~AttnFlatten.set_extra_state
      ~AttnFlatten.share_memory
      ~AttnFlatten.state_dict
      ~AttnFlatten.to
      ~AttnFlatten.to_empty
      ~AttnFlatten.train
      ~AttnFlatten.type
      ~AttnFlatten.xpu
      ~AttnFlatten.zero_grad
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~AttnFlatten.T_destination
      ~AttnFlatten.call_super_init
      ~AttnFlatten.dump_patches
      ~AttnFlatten.start_dim
      ~AttnFlatten.end_dim
      ~AttnFlatten.training