Pytorch中的数学函数

Posted 2021-01-01 sbj123456789

log_softmax

log(softmax(X))

function：torch.nn.functional.log_softmax(x, dim=None)

nn：torch.nn.LogSoftmax(dim=None)

如：

nll_loss

The negative log likelihood loss

function：torch.nn.functional.nll_loss(input, target, weight=None, size_average=True, ignore_index=-100, reduce=None, reduction=‘elementwise_mean‘)

nn：torch.nn.NLLLoss(weight=None, size_average=True, ignore_index=-100, reduce=None, reduction=‘elementwise_mean‘)

如：