交叉熵

交叉熵的公式：

 

p表示真实值，q表示预测值，H(p,q)表示交叉熵损失。

先对 inpu t求 softmax得到每个点的概率，然后在对结果求log，后面就是根据每个点的预测值以及真实值，按照上面的交叉熵公式计算。

torch.nn.functional.cross_entropy() 就是使用上式实现损失计算的。