skainet-lang-core/sk.ainet.lang.nn.hooks/TapeRecorder

TapeRecorder

class TapeRecorder : ForwardHooks(source)

Simple tape recorder that collects forward pass records. Zero-cost when not installed in the ExecutionContext (hooks == null).

Constructors

constructor()

Types

data class Entry(val moduleId: String, val moduleName: String, val modulePath: String?, val inputSpec: TensorSpec?, val outputSpec: TensorSpec?, val paramRefs: List<String>)

Properties

val tape: List<TapeRecorder.Entry>

Functions

open override fun onForwardBegin(module: ModuleNode, input: Any)

open override fun onForwardEnd(module: ModuleNode, input: Any, output: Any)