人工智能面试题更新 2026-08-05

请用 PyTorch 编写 SFT 训练中的损失计算代码,并说明为什么需要将标签序列右移一位。

字节跳动人工智能互联网/IT编码实现技术原理PyTorch

考察说明

考察 masked language modeling 损失实现与序列对齐理解

回答思路

  1. 正确实现 logits 与标签的移位对齐
  2. 正确构造 ignore_index 掩码处理填充位置
  3. 说明 shift right 的目的与计算细节
本题已收录答题指导

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

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