前向传播 Args: logits: 模型输出的未归一化对数概率 [batch_size, seq_len, vocab_size] labels: 真实词元索引 [batch_size, seq_len] prompt_lengths: 每个样本的 prompt 长度 [batch_size] Returns: loss: 标量损失值
(self, logits, labels, prompt_lengths)
source not stored for this graph (policy: none)
nothing calls this directly
no outgoing calls
no test coverage detected