蚂蚁集团面试题更新 2026-08-05
请用代码实现 decoder-only 架构中的因果注意力（causal attention）。
蚂蚁集团人工智能互联网/IT编码实现问题拆解技术原理
考察说明
考察因果掩码、缩放点积注意力和基础代码实现能力
回答思路
- 正确实现 Q、K、V 投影与点积
- 正确构造上三角掩码并处理负无穷填充
- 正确应用缩放因子与 softmax
- 写出可运行的简洁代码并解释掩码作用
本题附完整参考答案与评分标准
登录后可查看结构化答题指导；也可以直接开一场模拟面试，AI 面试官用本题实时追问并给出评分。
