← 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
Not currently ranked — collecting fresh signals.
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.

Frequently asked

What is mert-kurttutan/torchview?
A PyTorch model visualization library that renders neural network graphs showing layers, tensors, and input/output shapes.
Is torchview open source?
Yes — mert-kurttutan/torchview is open source, released under the MIT license.
What language is torchview written in?
mert-kurttutan/torchview is primarily written in Python.
How popular is torchview?
mert-kurttutan/torchview has 1.1k stars on GitHub.
Where can I find torchview?
mert-kurttutan/torchview is on GitHub at https://github.com/mert-kurttutan/torchview.

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