PyTorch の計算グラフ可視化

GitHub - szagoruyko/pytorchviz: A small package to create visualizations of PyTorch execution graphs でできそう。なんだかんだで計算グラフの観点から眺められると嬉しい。