skainet-lang-core/sk.ainet.lang.nn.activations/Softmax

Softmax

class Softmax<T : DType, V>(dimension: Int, val name: String = "Softmax") : Module<T, V> (source)

Constructors

constructor(dimension: Int, name: String = "Softmax")

Properties

open override val modules: List<Module<T, V>>

open override val name: String

Human-readable name of the node, typically the module's name.

Functions

open override fun forward(input: Tensor<T, V>, ctx: ExecutionContext): Tensor<T, V>