Attention
---------------------------------
.. currentmodule:: serket.nn
    
.. autoclass:: MultiHeadAttention