主页

索引

模块索引

搜索页面

概率分布-logsoftmax

• 用途: 用于数值稳定的分类问题。

• Softmax 的对数形式，常用于数值更稳定的交叉熵损失计算。

• 定义: $\text{logsoftmax}(x_i)=log(\frac{e^{x_i}}{\sum _{j=1}^n{e}^{x_j}})$

• 简化形式: $\text{logsoftmax}(x_i)=x_i-log(\sum _{j=1}^n{e}^{x_j})$

from torch.nn.functional import log_softmax
log_probs = log_softmax(x, dim=1)

主页

索引

模块索引

搜索页面