TensorFlow 2 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. |