腾讯面试题更新 2026-08-05

请手写实现一个Casual Self-Attention模块,说明其输入输出维度与计算流程。

腾讯人工智能互联网/IT编码实现技术原理PyTorch

考察说明

考察注意力机制的实现细节与因果掩码处理能力

回答思路

  1. 正确计算Q、K、V与缩放点积注意力
  2. 正确实现因果掩码防止未来信息泄露
  3. 说明输入输出维度及各组件关系
  4. 能指出数值稳定性处理(如softmax前掩码)
本题已收录答题指导

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

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