softmax
Softmax activation.
This layer applies softmax function to an input.
Parameter
- axis : Integer, axis along which the softmax normalization is applied.
The default is -1 (the last dimension).
Softmax activation.
This layer applies softmax function to an input.