一句话速记:单头的 softmax 一次只能聚焦少数位置,多头把 d 维拆成 h 个子空间,各自学不同的关系(语法、指代、位置),计算量基本不变。head_dim 一般固定 64 或 128,头数 = d / head_dim。
注意力