简介
tensorflow 图像分类问题中,常用的交叉熵损失计算有三个函数:
tf.nn.softmax_cross_entropy_with_logits
tf.nn.softmax_cross_entropy_with_logits_v2
tf.nn.sparse_softmax_cross_entropy_with_logits
简单记录一下这三个函数的用法,免得自己后面再忘记了。
1. softmax_cross_entropy_with_logits
该函数的声明如下:
def softmax_cross_entropy_with_logits(
_sentinel=None, # pylint: disable=invalid-name
labels=None,
log