PyTorch的NLLLoss和CrossEntropyLoss区别和联系

这个问题其他网友已经总结得很好，例如：https://blog.csdn.net/qq_22210253/article/details/85229988

为了方便大家更进一步理解，我把上面博客中的例子总结到了一个文件中，方便大家以此为例学习：

import torch
import torch.nn as nn

input=torch.randn(3,3)
print(input)

sm=nn.Softmax(dim=1)
print(sm(input))

log_result=torch.log(sm(input))
print(log_result)

###############################
loss=nn.NLLLoss()
target=torch.tensor([0,2,1])
input_sm=sm(input)
print(loss(torch.log(input_sm),target))
###############################
loss=nn.CrossEntropyLoss()
print(loss(input,target))