towhee.models.layers.attention

Classes

Attention	Scaled dot-product attention mechanism.
MultiHeadAttention	Multi-head attention layer.