skainet-lang-core/sk.ainet.lang.nn.dsl/NeuralNetworkDslImpl/softmax

softmax

open override fun softmax(dim: Int, id: String)(source)

Applies a Softmax activation as a separate layer.

Parameters

dim

Dimension along which softmax will be computed. Supports negative indexing.

id

Optional identifier for the layer