数值稳定的 Log Softmax 函数 使用 Log-Sum-Exp 技巧保证数值稳定性。 公式: log_softmax(x) = x - log(sum(exp(x))) Args: logits: 未归一化的对数概率 [batch_size, num_classes] Returns: log_softmax: 对数概率 [batch_size, num_classes]
(logits)
source not stored for this graph (policy: none)
no outgoing calls
no test coverage detected