Module attention

Source

Structsยง

MultiHeadAttention