交叉熵在TensorFlow中的计算

交叉熵作为一种loss函数在深度学习中可以作为梯度下降优化的对象,其计算公式为:
在这里插入图片描述
其中y是网络预测的概率分布,y‘则是真实测试数据的概率分布。
则对于二维tensor交叉熵的计算方式为:

cross_entropy = tf.reduce_mean(-tf.reduce_sum(tf.log(y)*y',axis=1))

猜你喜欢

转载自blog.csdn.net/cyr429/article/details/89244502
今日推荐