Module attention

Module attention 

Source

Structsยง

MultiHeadAttention
Multi-Head Self-Attention module.