TensorFlow函數(shù)教程：tf.keras.backend.categorical_crossentropy

tf.keras.backend.categorical_crossentropy函數(shù)

tf.keras.backend.categorical_crossentropy(
    target,
    output,
    from_logits=False,
    axis=-1
)

定義在：tensorflow/python/keras/backend.py。

輸出張量和目標(biāo)張量之間的分類交叉熵。

參數(shù)：

• target：與output具有相同形狀的張量。
• output：由softmax產(chǎn)生的張量（除非from_logits是True，在這種情況下output預(yù)計(jì)是logits）。
• from_logits：Boolean，判斷output是softmax的結(jié)果，還是logits的張量。
• axis：Int，用于指定通道軸。axis=-1對(duì)應(yīng)于數(shù)據(jù)格式channels_last', axis = 1對(duì)應(yīng)于數(shù)據(jù)格式channels_first'。

返回：

輸出張量。

可能引發(fā)的異常：

• ValueError：如果axis既不是-1，也不是output的其中一個(gè)軸。