Skip to main content

Module visualization

Module visualization 

Source
Expand description

Visualization utilities for computation graphs and proofs.

This module provides tools for exporting graphs and proofs to DOT format (Graphviz) for visualization and debugging.

§Examples

§Visualizing a Computation Graph

use ipfrs_tensorlogic::{ComputationGraph, GraphNode, TensorOp, GraphVisualizer};

let mut graph = ComputationGraph::new();

// Create nodes
let input = GraphNode::new("input".to_string(), TensorOp::Input {
    name: "x".to_string(),
});
graph.add_node(input).expect("write to String is infallible");
graph.mark_input("input".to_string());

let relu = GraphNode::new("relu".to_string(), TensorOp::ReLU)
    .add_input("input".to_string());
graph.add_node(relu).expect("write to String is infallible");
graph.mark_output("relu".to_string());

// Export to DOT format
let dot = GraphVisualizer::to_dot(&graph);
println!("{}", dot);
// Save to file: std::fs::write("graph.dot", dot).expect("write to String is infallible");
// Render: dot -Tpng graph.dot -o graph.png

Structs§

GraphVisualizer
Visualizer for computation graphs
ProofVisualizer
Visualizer for proof trees