TensorFlow函數(shù)教程：tf.nn.softmax_cross_entropy_with_logits

tf.nn.softmax_cross_entropy_with_logits函數(shù)

tf.nn.softmax_cross_entropy_with_logits(
    _sentinel=None,
    labels=None,
    logits=None,
    dim=-1,
    name=None
)

定義在：tensorflow/python/ops/nn_ops.py。

計(jì)算logits和labels之間的softmax交叉熵。（廢棄）

這個(gè)函數(shù)已被棄用。它將在以后的版本中刪除。更新說(shuō)明：TensorFlow的未來(lái)主要版本將允許漸變?cè)谀J(rèn)情況下流入backprop上的標(biāo)簽輸入。

參考：tf.nn.softmax_cross_entropy_with_logits_v2。

測(cè)量離散分類任務(wù)中的概率誤差，其中類是互斥的（每個(gè)條目恰好在一個(gè)類中）。 例如，每個(gè)CIFAR-10圖像都標(biāo)有一個(gè)且只有一個(gè)標(biāo)簽：圖像可以是dog或truck，但不能同時(shí)是兩者。

注意：雖然這些類是互斥的，但它們的概率不一定如此。所需要的只是每一行l(wèi)abels是有效的概率分布。如果不是，則梯度的計(jì)算將是不正確的。

如果使用專屬labels（其中一次只有一個(gè)類為true），請(qǐng)參閱sparse_softmax_cross_entropy_with_logits。

警告：此操作需要未縮放的日志，因?yàn)樗趦?nèi)部的logits執(zhí)行l(wèi)ogmax以提高效率。不要使用softmax的輸出調(diào)用此操作，因?yàn)樗鼤?huì)產(chǎn)生不正確的結(jié)果。

常見(jiàn)的用例是具有shape [batch_size，num_classes]的logits和labels，但支持更高的維度，dim參數(shù)指定類維度。

反向傳播只會(huì)發(fā)生在logits中。要計(jì)算允許反向傳播到logits和labels的交叉熵?fù)p失，請(qǐng)參閱tf.nn.softmax_cross_entropy_with_logits_v2。

請(qǐng)注意，為避免混淆，需要僅將命名參數(shù)傳遞給此函數(shù)。

參數(shù)：

• _sentinel：用于防止positional參數(shù)。內(nèi)部的，不要使用。
• labels：沿著類維度的每個(gè)向量應(yīng)該保持有效的概率分布，例如：對(duì)于標(biāo)簽shape為[batch_size, num_classes]的情況，labels[i]的每一行必須是有效的概率分布。
• logits：未縮放的日志概率。
• dim：類維度。默認(rèn)為-1，這是最后一個(gè)維度。
• name：操作的名稱（可選）。

返回：

一個(gè)Tensor，包含softmax交叉熵?fù)p失的。它的類型與logits相同，它的shape與labels是相同的，除了它沒(méi)有l(wèi)abels的最后一個(gè)維度。