NLLLoss

    该接口可创建一个NLLLoss可调用类,计算输入x和标签label间的 negative log likelihood loss 损失 ,可用于训练一个 n 类分类器。

    如果提供 weight 参数的话,它是一个 1-D 的tensor, 里面的值对应类别的权重。当你的训练集样本 不均衡的话,使用这个参数是非常有用的。

    该损失函数的数学计算公式如下:

    当 reduction 设置为 none 时,损失函数的数学计算公式为:

    其中 N 表示 batch_size 。如果 reduction 的值不是 none (默认为 mean),那么此时损失函数 的数学计算公式为: