skainet-compile-core/sk.ainet.compile.nn/NeuralNetworkContext

NeuralNetworkContext

interface NeuralNetworkContext<T : DType, V>(source)

Context for the DSL to define the data type and operations.

This class holds the information about the data type and operations that should be used in the DSL. It's used to make the DSL generic and to avoid hardcoding the data type.

Parameters

The default data type.

Inheritors

DefaultNetworkContext

Properties

executionContext

abstract val executionContext: NeuralNetworkExecutionContext

Functions

network

inline fun <T : DType, V> NeuralNetworkContext<T, V>.network(content: NeuralNetworkDsl<T, V>.() -> Unit): Module<T, V>

Extension function to create a network within a NetworkContext. This bridges the context wrapper with the network DSL using the context's tensor factory.