Module attention

Module attention 

Source

Structsยง

MultiHeadAttention