transformer/multi_head_cross_attention library

Classes

MultiHeadCrossAttention
Combines multiple cross-attention heads to run in parallel.