请手写实现多头注意力机制（Multi-head Attention）的核心部分。

平安产险科技中心人工智能金融编码实现问题拆解技术原理PyTorch

考察说明

考察对Transformer注意力机制的理解与代码实现能力

本题已收录答题指导

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