请手写实现多头自注意力（MHSA）的核心代码，并说明关键步骤。

Question

Accepted Answer

考察点：考察对多头自注意力机制的理解及代码实现能力。回答思路：正确实现Q、K、V的线性投影与维度变换 正确切分多头并计算缩放点积注意力 正确拼接输出并进行输出投影 考察对mask、dropout等细节的处理

考察说明