请手写并解释多头注意力（Multi-Head Attention）机制的代码实现。

Question

Accepted Answer

考察点：考察多头注意力的实现细节与原理理解。回答思路：正确实现缩放点积注意力及掩码处理 正确划分并拼接多头，维度计算无误 能解释每个步骤的作用和复杂度 能说明多头注意力相对单头注意力的优势

考察说明