← all repositories

mert-kurttutan/torchview

A PyTorch model visualization library that renders neural network graphs showing layers, tensors, and input/output shapes.

1.1k stars Python ML Frameworks
torchview
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

Torchview provides visual graph representations of PyTorch models, functioning as a PyTorch equivalent to keras plot_model. It renders neural network architectures including tensors, modules, torch functions, and dimension information. The tool uses graphviz to generate the visual graphs and supports PyTorch versions 1.7 and above.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.