softmax_classification_output

tfsnippet.ops.softmax_classification_output(logits, name=None)

Get the most possible softmax classification output for each logit.

Parameters:

• logits – The softmax logits. Its last dimension will be treated as the softmax logits dimension, and will be reduced.
• name (str) – Default name of the name scope. If not specified, generate one according to the method name.

Returns:

tf.int32 tensor, the class label for each logit.

Return type:

tf.Tensor