TensorFlow 1 version
View source on GitHub
Softmax activation function.
Inherits From: Layer
View aliases
Compat aliases for migration
See Migration guide for more details.
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.
Arguments | |
|---|---|
axis
|
Integer, axis along which the softmax normalization is applied. |