W3cubDocs

tf.raw_ops.Softmax

Computes softmax activations.

Compat aliases for migration

tf.compat.v1.raw_ops.Softmax

tf.raw_ops.Softmax(
    logits, name=None
)

For each batch i and class j we have

$$softmax[i, j] = exp(logits[i, j]) / sum_j(exp(logits[i, j]))$$

Args
`logits`	A `Tensor`. Must be one of the following types: `half`, `bfloat16`, `float32`, `float64`. 2-D with shape `[batch_size, num_classes]`.
`name`	A name for the operation (optional).

Returns
A `Tensor`. Has the same type as `logits`.

© 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/r2.4/api_docs/python/tf/raw_ops/Softmax