skainet-lang-core/sk.ainet.lang.nn.topology/ModuleNode

ModuleNode

A generic, dtype-agnostic node interface for traversing module hierarchies. This allows programmatic traversal, parameter collection, and introspection without depending on the input/output dtype generics of Module.

Inheritors

DualModule

Module

ModuleNodeAdapter

Properties

children

abstract val children: List<ModuleNode>

Child nodes in the hierarchy.

abstract val id: String

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

name

abstract val name: String

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

params

abstract val params: List<ModuleParameter<*, *>>

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

path

abstract var path: String?

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

Functions

collectParams

fun ModuleNode.collectParams(): List<ModuleParameter<*, *>>

findById

fun ModuleNode.findById(id: String): ModuleNode?

findByPath

fun ModuleNode.findByPath(path: String): ModuleNode?

walkDepthFirst

fun ModuleNode.walkDepthFirst(visit: (ModuleNode) -> Unit)

Traversal utilities for ModuleNode hierarchies.