Struct GradientFlowVisualizer 

pub struct GradientFlowVisualizer<T> { /* private fields */ }

Gradient flow visualizer for computation graphs

Implementations

impl<T> GradientFlowVisualizer<T>

where T: Float + Clone + Default + Zero + One + Send + Sync + 'static + Add<Output = T> + Sub<Output = T> + Mul<Output = T> + Div<Output = T> + Neg<Output = T> + PartialOrd + FromPrimitive + Signed + Debug + Serialize + DeserializeOwned + Pod + Zeroable,

pub fn new() -> Self

Create a new gradient flow visualizer

pub fn with_settings(settings: VisualizationSettings) -> Self

Create visualizer with custom settings

pub fn settings(&self) -> &VisualizationSettings

Get the current visualization settings

pub fn set_settings(&mut self, settings: VisualizationSettings)

Update visualization settings

pub fn nodes(&self) -> &HashMap<TensorId, GradientFlowNode<T>>

Get the nodes in the visualization graph

pub fn edges(&self) -> &[GradientFlowEdge]

Get the edges in the visualization graph

pub fn flow_analysis(&self) -> Option<&GradientFlowAnalysis<T>>

Get the flow analysis results

pub fn analyze_flow( &mut self, tape: &GradientTape, target: &TrackedTensor<T>, sources: &[&TrackedTensor<T>], ) -> Result<()>

Analyze gradient flow from a computation graph

pub fn clear(&mut self)

Clear all analysis results

pub fn get_health_summary(&self) -> Result<String>

Get a summary of the gradient flow health

Trait Implementations

impl<T: Clone> Clone for GradientFlowVisualizer<T>

fn clone(&self) -> GradientFlowVisualizer<T>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: Debug> Debug for GradientFlowVisualizer<T>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<T> Default for GradientFlowVisualizer<T>

where T: Float + Clone + Default + Zero + One + Send + Sync + 'static + Add<Output = T> + Sub<Output = T> + Mul<Output = T> + Div<Output = T> + Neg<Output = T> + PartialOrd + FromPrimitive + Signed + Debug + Serialize + DeserializeOwned + Pod + Zeroable,

fn default() -> Self

Returns the “default value” for a type.