tf.keras.activations.softmax

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

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

Softmax activation function.

Arguments:

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

Returns:

Tensor, output of softmax transformation.

Raises:

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