skainet-lang-core/sk.ainet.context/ExecutionContext

ExecutionContext

interface ExecutionContext(source)

Inheritors

DefaultDataExecutionContext

PhaseOverridingExecutionContext

TrainingExecutionContext

NeuralNetworkExecutionContext

Properties

abstract val executionStats: ExecutionStats

open val hooks: ForwardHooks?

open val inTraining: Boolean

abstract val memoryInfo: MemoryInfo

abstract val observers: ExecutionObserverRegistry

abstract val ops: TensorOps

abstract val phase: Phase

tensorDataFactory

abstract val tensorDataFactory: TensorDataFactory

Functions

open fun <T : DType, V> fromByteArray(shape: Shape, dtype: KClass<T>, data: ByteArray): Tensor<T, V>

open fun <T : DType, V> fromData(data: TensorData<T, V>, dtype: KClass<T>): Tensor<T, V>

open fun <T : DType, V> fromFloatArray(shape: Shape, dtype: KClass<T>, data: FloatArray): Tensor<T, V>

open fun <T : DType, V> fromIntArray(shape: Shape, dtype: KClass<T>, data: IntArray): Tensor<T, V>

open fun <T : DType, V> full(shape: Shape, dtype: KClass<T>, value: Number): Tensor<T, V>

open fun <T : DType, V> ones(shape: Shape, dtype: KClass<T>): Tensor<T, V>

registerObserver

open fun registerObserver(observer: ExecutionObserver)

unregisterObserver

open fun unregisterObserver(observer: ExecutionObserver)

open fun <T : DType, V> zeros(shape: Shape, dtype: KClass<T>): Tensor<T, V>