skainet-lang-core/sk.ainet.lang.trace/TraceSession

TraceSession

TraceSession manages tensor references during computation graph tracing. It provides methods to convert tensors to TensorRef objects for graph capture.

Constructors

TraceSession

constructor()

Functions

clear

fun clear()

Clear all cached tensor references. Call between training batches to prevent memory accumulation from intermediate tensors. Tensors will be re-registered on the next forward pass.

refOf

open fun refOf(tensor: Tensor<*, *>): TensorRef

Get or create a TensorRef for the given tensor.

refsOf

fun refsOf(tensors: List<Tensor<*, *>>): List<TensorRef>

Convert a list of tensors to TensorRef objects.

resolve

open fun resolve(id: String): Tensor<*, *>?

Resolve a tensor ID back to its original Tensor.

open fun resolve(ref: TensorRef): Tensor<*, *>?

Resolve a TensorRef back to its original Tensor.