我要招人

人工智能面试题更新 2026-08-05

请编写一个多头注意力机制的代码实现。

旷视人工智能编码实现技术原理PyTorchTransformer

考察说明

考察对注意力机制原理的理解及编程实现能力

回答思路

  1. 正确实现Q、K、V线性投影及按头拆分
  2. 实现缩放点积注意力并计算注意力权重
  3. 能够合并多头的输出并投影回原始维度
  4. 代码结构清晰、变量命名规范、无明显逻辑错误
本题已收录答题指导

本题附完整参考答案与评分标准

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

开始模拟面试登录查看答题指导