tf.keras.layers.Softmax

TensorFlow 2 version View source on GitHub

Softmax activation function.

Inherits From: Layer

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.layers.Softmax, `tf.compat.v2.keras.layers.Softmax`

 
tf.keras.layers.Softmax(
    axis=-1, **kwargs
)

Input shape:

Arbitrary. Use the keyword argument input_shape (tuple of integers, does not include the samples axis) when using this layer as the first layer in a model.

Output shape:

Same shape as the input.

axis Integer, axis along which the softmax normalization is applied.