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

2019-03-07 16:51 更新

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的其中一个轴。
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号