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

softmax

abstract fun softmax(dim: Int = -1, 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