torchviz

A small package to create visualizations of PyTorch execution graphs


License
BSD-3-Clause
Install
pip install torchviz==0.0.2