Missed TensorFlow World? Check out the recap. Learn more

tf.keras.backend.sparse_categorical_crossentropy

TensorFlow 2.0 version View source on GitHub

Categorical crossentropy with integer targets.

Aliases:

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

Arguments:

  • target: An integer tensor.
  • output: A tensor resulting from a softmax (unless from_logits is True, in which case output is expected to be the logits).
  • from_logits: Boolean, whether output is the result of a softmax, or is a tensor of logits.
  • axis: Int specifying the channels axis. axis=-1 corresponds to data format channels_last', andaxis=1corresponds to data formatchannels_first`.

Returns:

Output tensor.

Raises:

  • ValueError: if axis is neither -1 nor one of the axes of output.