tf.keras.activations.softmax

tf.keras.activations.softmax(
    x,
    axis=-1
)

Defined in tensorflow/python/keras/activations.py.

Softmax activation function.

Arguments:

• x: Input tensor.
• axis: Integer, axis along which the softmax normalization is applied.

Returns:

Tensor, output of softmax transformation.

Raises:

• ValueError: In case dim(x) == 1.