tf.keras.backend.sparse_categorical_crossentropy
View source on GitHub |
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`. |
Returns | |
---|---|
Output tensor. |
Raises | |
---|---|
ValueError |
if axis is neither -1 nor one of the axes of output . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/backend/sparse_categorical_crossentropy