RSVP for your your local TensorFlow Everywhere event today!


TensorFlow 1 version View source on GitHub

Categorical crossentropy with integer targets.

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`.

Output tensor.

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