百度面试题更新 2026-08-05

请手写实现 Multi-Head Attention(MHA),并说明其计算过程。

百度人工智能专业服务编码实现技术原理PyTorch

考察说明

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

回答思路

  1. 正确实现 Q、K、V 的线性变换与多头拆分
  2. 正确计算缩放点积注意力并应用 softmax
  3. 正确拼接多头输出并做最终线性投影
  4. 能说明维度变化与计算细节
本题已收录答题指导

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

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