请手写多头自注意力机制的实现，并说明其工作原理。

Question

Accepted Answer

考察点：考察多头自注意力的实现细节与核心概念理解。回答思路：能独立写出Q、K、V线性变换与多头拆分 正确处理缩放点积注意力及softmax计算 能合并多头输出并经过输出投影 口头解释多头机制的作用与优势 能指出维度匹配和注意力掩码等要点

考察说明