ActivationsWrapperModule

class ActivationsWrapperModule<T : DType, V>(activationHandler: (Tensor<T, V>) -> Tensor<T, V>, val name: String) : Module<T, V> (source)

Constructors

constructor(activationHandler: (Tensor<T, V>) -> Tensor<T, V>, name: String)

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

open override val name: String

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

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