摘要
本文使用纯 Python 和 PyTorch 对比实现cross-entropy交叉熵损失函数及其反向传播.
相关
原理和详细解释, 请参考文章 :
通过案例详解cross-entropy交叉熵损失函数
系列文章索引 :
https://blog.csdn.net/oBrightLamp/article/details/85067981
正文
1. 定义:
E = − ∑ i = 1 k y i l o g ( s i ) E = -\sum_{i = 1}^{k}y_{i}log(s_{i})\\ E=−i=1∑kyilog(si)
2. 梯度:
▽ E ( s ) = ( ∂ E ∂ s 1 , ∂ E ∂ s 2 , ⋯   , ∂ E ∂