我要招人

字节跳动面试题更新 2026-08-05

请手写实现多头注意力的核心计算过程。

字节跳动人工智能互联网/IT编码实现技术原理PyTorch

考察说明

考察对自注意力机制、多头拆分与拼接的编码实现能力

回答思路

  1. 正确实现 Q、K、V 线性投影与多头拆分
  2. 正确计算缩放点积注意力并应用掩码
  3. 正确拼接各头输出并做最终投影
  4. 注意维度变换与转置的边界
  5. 能解释各步骤的作用
本题已收录答题指导

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

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

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