pytorchviz icon indicating copy to clipboard operation
pytorchviz copied to clipboard

Published 20 hours ago • verified publisher icon szagoruyko

Update dot.py

Open z-huabao opened this issue 4 years ago • 1 comments

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 的屏幕截图

z-huabao avatar Jul 12 '19 09:07 z-huabao

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

lizhaoliu-Lec avatar Mar 26 '20 16:03 lizhaoliu-Lec