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

tf.keras.backend.categorical_crossentropy函数

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

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

输出张量和目标张量之间的分类交叉熵。

参数：

• target：与output具有相同形状的张量。
• 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的其中一个轴。