拖动 dk,看 logits 怎么变宽、softmax 怎么变尖、梯度怎么塌。全部现算,不查表。
单个 logit = q·k。理论标准差:不缩放 √dk,缩放后 1。
同一组 logits 归一化后的权重分布。看它有多接近 one-hot。
Jij = pi(δij − pj)。这个范数越小,误差信号越难穿过 softmax 往回传。