skainet-lang-core/sk.ainet.lang.nn/DualModule

DualModule

abstract class DualModule<InT : DType, OutT : DType, V> : ModuleNode(source)

Dual-typed module for dtype-transforming or index-consuming ops. InT is the input dtype, OutT is the output dtype. V is the value buffer backend type.

Inheritors

Embedding

Constructors

DualModule

constructor()

Properties

children

open override val children: List<ModuleNode>

Child nodes in the hierarchy.

open override val id: String

A stable identifier for the node. Not required to be globally unique.

modules

abstract val modules: List<ModuleNode>

Child modules/nodes for traversal. Keep dtype-agnostic.

name

abstract override val name: String

Human-readable module name

params

open override val params: List<ModuleParameter<*, *>>

Parameters owned by this node (weights, biases, etc.).

path

open override var path: String?

Optional path annotation for nicer logs, e.g., "model.encoder.layer1".

Functions

forward

abstract fun forward(input: Tensor<InT, V>, ctx: ExecutionContext): Tensor<OutT, V>

Forward pass that requires an ExecutionContext.