Struct TensorInspector 

pub struct TensorInspector { /* private fields */ }

Tensor inspector for examining intermediate values.

Implementations

impl TensorInspector

pub fn new() -> Self

Create a new tensor inspector.

pub fn enable(&mut self)

Enable inspection.

pub fn disable(&mut self)

Disable inspection.

pub fn is_enabled(&self) -> bool

Check if inspection is enabled.

pub fn watch(&mut self, tensor_id: usize)

Add a tensor to the watch list.

pub fn unwatch(&mut self, tensor_id: usize)

Remove a tensor from the watch list.

pub fn clear_watch_list(&mut self)

Clear the watch list.

pub fn should_inspect(&self, tensor_id: usize) -> bool

Check if a tensor should be inspected.

pub fn record_stats(&mut self, stats: TensorStats)

Record tensor statistics.

pub fn get_stats(&self, tensor_id: usize) -> Option<&TensorStats>

Get statistics for a tensor.

pub fn get_all_stats(&self) -> &HashMap<usize, TensorStats>

Get all recorded statistics.

pub fn find_problematic_tensors(&self) -> Vec<&TensorStats>

Find tensors with numerical issues.

pub fn clear(&mut self)

Clear all recorded statistics.

Trait Implementations

impl Default for TensorInspector

fn default() -> Self

Returns the “default value” for a type.