Is there a way to generate and iterate successively through a computation graph in PyTorch, similar
to how Tensorflow returns a Graph (or GraphDef) object?
I understand you can use torchviz/graphviz to visualize the computation graph, but is there a way to get the graph in the finest granularity possible, i.e. the individual mult/add/etc. ops?
thanks!