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

阿里巴巴人工智能电商编码实现技术原理PyTorch

考察说明

考察多头注意力的实现细节与原理理解

本题已收录答题指导

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