请编写多头自注意力模块的前向传播（forward）实现，并说明关键步骤。

北京智谱华章科技人工智能互联网/IT编码实现技术原理PyTorch

考察说明

考察注意力机制实现与关键细节

本题已收录答题指导

登录后可查看结构化答题指导；也可以直接开一场模拟面试，AI 面试官用本题实时追问并给出评分。