TensorFlow函數(shù)教程:tf.keras.backend.sparse_categorical_crossentropy

2019-04-14 20:31 更新

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

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

定義在:tensorflow/python/keras/backend.py

具有整數(shù)目標(biāo)的分類交叉熵。

參數(shù):

  • target:整數(shù)張量。
  • output:由softmax產(chǎn)生的張量(除非from_logits是True,在這種情況下output預(yù)計(jì)是logits)。
  • from_logits:Boolean,無(wú)論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è)軸。


以上內(nèi)容是否對(duì)您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)