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', andaxis=1corresponds to data formatchannels_first`. |
| Raises |
ValueError |
if axis is neither -1 nor one of the axes of output. |