TensorFlow 1 version | View source on GitHub |
Softmax activation function.
Inherits From: Layer
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. |