torchvista：用一行代码在Jupyter等笔记本中直观可视化PyTorch模型的前向传播过程。

交互式图形支持拖拽和缩放；支持层级模块的可折叠节点；错误容忍的局部可视化功能，方便调试
 
 
Back to Top