tf.keras.losses.sparse_categorical_crossentropy

View source on GitHub

tf.keras.losses.sparse_categorical_crossentropy(
    y_true, y_pred, from_logits=False, axis=-1
)