Solved by albanD in post #2 
              
              
                Hi, 
You can use this script to create a graph https://github.com/szagoruyko/functional-zoo/blob/master/visualize.py 
To open it with a regular pdf viewer you can do make_dot(your_var).view().