ยป Keras API reference / Layers API / Activation layers / Softmax layer

Softmax layer

Softmax class

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

Softmax activation function.

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.

Arguments

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