注意力机制

多头注意力机制

多头注意力的计算流程、维度变化及 Flash Attention 优化原理。