先放上官方用法

torch.nn.functional.nll_loss(input, target, weight=None, size_average=None, ignore_index=-100, reduce=None, reduction='mean')

一般只赋值前两个张量就可以了。