TensorFlow Memo: 两个交叉熵

API - 1:

tf.nn.sparse_softmax_cross_entropy_with_logits(logits, labels)

它的 labels 是类别的一维数组, 如 [0, 3, 9, 4, 8, 9]

API - 2:

tf.nn.softmax_cross_entropy_with_logits_v2(logits, labels)

它的 label 是 one hot 表示, 如
[[0, 0, 0, 1], [0, 0, 1, 0], [0, 0, 0, 1], [1, 0, 0, 0]]

猜你喜欢

转载自blog.csdn.net/vinceee__/article/details/88410300