TensorFlow函数教程：tf.keras.backend.sparse_categorical_crossentropy

tf.keras.backend.sparse_categorical_crossentropy函数

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

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

具有整数目标的分类交叉熵。

参数：

• target：整数张量。
• output：由softmax产生的张量（除非from_logits是True，在这种情况下output预计是logits）。
• from_logits：Boolean，无论output是softmax的结果，还是logits的张量。
• axis：Int指定通道轴。axis=-1对应于数据格式channels_last'，并且axis = 1对应于数据格式channels_first`。

返回：

输出张量。

可能引发的异常：

• ValueError：如果axis既不是-1也不是output中的某一个轴。