Skip to content

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).