Struct StepExecutor 

pub struct StepExecutor<E> {
    pub inner: E,
    pub log: Vec<IntermediateValue>,
    /* private fields */
}

Wraps any TlExecutor and logs IntermediateValue snapshots at each operation.

A snapshot is recorded when at least one active BreakpointCondition triggers. If no conditions are added no logging occurs; add BreakpointCondition::Always to capture every step.

Fields

inner: E

The inner executor that performs actual computation.

log: Vec<IntermediateValue>

Accumulated log of intermediate values.

Implementations

impl<E> StepExecutor<E>

pub fn new(inner: E) -> Self

Create a new StepExecutor wrapping inner with no active conditions.

pub fn add_condition(&mut self, cond: BreakpointCondition)

Add a breakpoint condition.

pub fn log(&self) -> &[IntermediateValue]

View the accumulated log.

pub fn step_count(&self) -> usize

Total number of operations executed so far.

pub fn clear_log(&mut self)

Clear the accumulated log (step count is not reset).

pub fn has_nan_in_log(&self) -> bool

Returns true if any logged entry contains NaN.

pub fn has_inf_in_log(&self) -> bool

Returns true if any logged entry contains Inf.

pub fn summary(&self) -> String

One-line human-readable summary of the execution log.

Trait Implementations

impl<E> TlExecutor for StepExecutor<E>

where E: TlExecutor<Tensor = ArrayD<f64>>,

TlExecutor implementation for executors whose tensor type is ArrayD<f64>.

type Tensor = ArrayBase<OwnedRepr<f64>, Dim<IxDynImpl>>

type Error = <E as TlExecutor>::Error

fn einsum( &mut self, spec: &str, inputs: &[Self::Tensor], ) -> Result<Self::Tensor, Self::Error>

Execute an einsum operation on input tensors.

fn elem_op( &mut self, op: ElemOp, x: &Self::Tensor, ) -> Result<Self::Tensor, Self::Error>

Apply an element-wise unary operation.

fn elem_op_binary( &mut self, op: ElemOp, x: &Self::Tensor, y: &Self::Tensor, ) -> Result<Self::Tensor, Self::Error>

Apply an element-wise binary operation.

fn reduce( &mut self, op: ReduceOp, x: &Self::Tensor, axes: &[usize], ) -> Result<Self::Tensor, Self::Error>

Reduce a tensor along specified axes.