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

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.