TrainingExecutionContext

Extension of ExecutionContext that supports gradient recording and backward pass. This interface allows high-level training utilities to remain backend-agnostic while leveraging autograd capabilities when available.

Functions

Link copied to clipboard
abstract fun backward(targets: List<Tensor<*, *>>, sources: List<Tensor<*, *>>)

Perform backward pass from targets to sources. Populates Tensor.grad for involved tensors.

Link copied to clipboard
abstract fun startRecording()

Start recording operations for autograd.

Link copied to clipboard
abstract fun stopRecording(): Any?

Stop recording operations.