search

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

Update dot.py #33

Open z-huabao opened 5 years ago

z-huabao commented 5 years ago

1, remove Backend of node(inner tensor) name 2, add shape to the node 3, add input and input shape 4, change make_dot function args to make_dot(model, x)

2019-07-12 17-38-04 的屏幕截图

lizhaoliu-Lec commented 4 years ago

The new features added by you is fantastic!!! Can you find a way to support multiple inputs? Thanks very much!