AI 面试手写代码题:Multi-head Attention 到 LRU 怎么练
2026-07-15 · 博客首页
AI 岗位的编程考察分两层:通用算法题(和所有程序员一样)+ AI 特有的手写题。后者是拉开差距的地方。
AI 特有的手写题清单
必练五个:
1. 手写 Multi-head Attention——考得最多,公式到代码一步不能错;
2. 手写 LayerNorm / BatchNorm——注意维度和训练/推理差异;
3. 手写 KMeans——考察基本实现能力;
4. 手写 Beam Search——生成方向常考;
5. 手写 IoU / NMS——CV 方向必考。
练习方法:每个先自己写,卡住截图问 AI 拿参考实现,然后关掉一切默写——面试时没有补全提示,肌肉记忆才可靠。
通用算法题
大厂照常考 LeetCode 中等难度:数组、链表、二叉树、动态规划是主力。刷题姿势(先想 15 分钟 → 截图问 AI → 关掉默写)在算法题那篇讲过,不重复。
现场发挥三技巧
- 先说思路再写:面试官点头了再动手,方向错写再多白费;
- 边界条件主动提:"这里我假设输入非空,如果是空的我会……"——展示工程素养;
- 写完跑个例子:口头代入一组数据走一遍,自己抓 bug 比面试官指出来体面得多。
备考节奏
两周:AI 手写题 5 个各默写 3 遍 + 通用算法每天 3 题。题解全部截图问答搞定,工具免费试 3 次。