互联网/IT行业面试题更新 2026-08-05
请手写实现一个标准的Scaled Dot-Product Attention，并说明其计算流程。
字节跳动小红书人工智能互联网/IT专业服务编码实现技术原理PyTorch
考察说明
考察对Transformer核心注意力机制的实现理解与推导能力
回答思路
- 能写出Q、K、V点积及缩放因子1/√d_k的计算
- 能解释softmax归一化及mask的作用
- 能说明计算复杂度与memory占用
- 能引申多头注意力的拼接投影逻辑
本题附完整参考答案与评分标准
登录后可查看结构化答题指导；也可以直接开一场模拟面试，AI 面试官用本题实时追问并给出评分。
