先放上官方用法
torch.nn.functional.nll_loss(input, target, weight=None, size_average=None, ignore_index=-100, reduce=None, reduction='mean')
torch.nn.functional.
nll_loss
一般只赋值前两个张量就可以了。