bapred.model.MHA.MultiHeadAttention
===================================

.. currentmodule:: bapred.model.MHA

.. autoclass:: MultiHeadAttention

   
   .. automethod:: __init__

   
   .. rubric:: Methods

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

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~MultiHeadAttention.T_destination
      ~MultiHeadAttention.call_super_init
      ~MultiHeadAttention.dump_patches
      ~MultiHeadAttention.training