昆仑万维面试题更新 2026-08-05
请手写实现一个 multi-head attention 模块，并说明其计算过程。
昆仑万维人工智能游戏编码实现问题拆解技术原理PyTorch
考察说明
考察对 multi-head attention 原理与实现的掌握，以及代码正确性与清晰度
回答思路
- 正确计算 Q、K、V 线性变换并切分多头
- 正确进行缩放点积注意力计算并应用 mask
- 正确拼接多头输出并做输出投影
- 代码简洁可运行，能解释维度变化
本题附完整参考答案与评分标准
登录后可查看结构化答题指导；也可以直接开一场模拟面试，AI 面试官用本题实时追问并给出评分。
