search

mert-kurttutan / torchview

torchview: visualize pytorch models
https://torchview.dev
MIT License
824 stars 38 forks source link

Show only outer-most output node #45

Closed mert-kurttutan closed 1 year ago