tf.keras.backend.sparse_categorical_crossentropy
Categorical crossentropy with integer targets.
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', and axis=1corresponds to data format channels_first`. |
Raises |
ValueError | if axis is neither -1 nor one of the axes of output . |