函数说明
在计算交叉熵之前,通常要用到softmax层来计算结果的概率分布。因为softmax层并不会改变最终的分类结果(排序),所以,tensorflow将softmax层与交叉熵函数进行封装,形成一个函数方便计算:
tf.nn.softmax_cross_entropy_with_logits(logits= , labels=)
。
为了加速计算过程,针对只有一个正确答案(例如MNIST识别)的分类问题,tensorflow提供了
tf.nn.sparse_softmax_cross_entropy_with_logits(logits= , labels=)
。