Struct GradientFlowVisualizer 

pub struct GradientFlowVisualizer { /* private fields */ }

Gradient flow visualizer 勾配フロービジュアライザー

Implementations

impl GradientFlowVisualizer

pub fn new() -> Self

Create a new gradient flow visualizer 新しい勾配フロービジュアライザーを作成

pub fn trace_from_variable<T>( &mut self, var: &Variable<T>, label: &str, ) -> usize

where T: Float + Debug + Send + Sync + 'static + ScalarOperand + FromPrimitive,

Trace gradient flow from a variable 変数から勾配フローをトレース

pub fn add_operation( &mut self, op_name: &str, inputs: Vec<usize>, output: usize, )

Add an operation node 演算ノードを追加

pub fn to_dot(&self) -> String

Generate DOT format for Graphviz Graphviz用のDOT形式を生成

pub fn save_to_file(&self, path: &Path) -> Result<()>

Save visualization to file 可視化をファイルに保存

pub fn gradient_flow_summary(&self) -> GradientFlowSummary

Generate a summary of gradient flow statistics 勾配フロー統計のサマリーを生成

pub fn detect_issues(&self) -> Vec<GradientFlowIssue>

Detect potential gradient flow issues 潜在的な勾配フローの問題を検出

pub fn clear(&mut self)

Clear the visualizer for reuse 再利用のためにビジュアライザーをクリア

Trait Implementations

impl Default for GradientFlowVisualizer

fn default() -> Self

Returns the “default value” for a type.