Module visualization 

Gradient Flow Visualization Data Generation

This module provides comprehensive visualization data generation for gradient flow analysis, including network topology, temporal flows, and critical path identification.

Structs

CriticalGradientPath

Critical path in gradient flow

ExplodingRegion

Region where gradients are exploding

FlowEdge

Edge in the gradient flow network

FlowNode

Node in the gradient flow network

GradientDeadZone

Zone where gradient flow is effectively dead

GradientDirection

Direction and characteristics of gradient flow

GradientFlowNetwork

Network representation of gradient flows

GradientFlowVisualization

Complete gradient flow visualization data

GradientFlowVisualizer

Gradient flow visualization generator

GradientLayerFlow

Gradient flow data for a specific layer

GradientVisualizationConfig

Configuration for gradient visualization

NetworkMetrics

Network-level gradient flow metrics

RegionExtent

Spatial extent of a gradient region

TemporalGradientFlow

Temporal gradient flow at a specific timestep

VanishingRegion

Region where gradients are vanishing

Enums

ColorScheme

EdgeType

Type of edge in the flow network

ExplodingSeverity

FlowDirection

Flow direction classification

NodeType

Type of node in the flow network

RecoveryPotential

VanishingSeverity