szagoruyko/pytorchviz

A small package to create visualizations of PyTorch execution graphs

语言：Jupyter Notebook · GitHub 星标：3,504 · Fork：290

在 GitHub 上查看原仓库