DasPantom:

I am trying to write a Function for torch.trace (seems not to exist so far). Here is my code:


I am not sure if this was different in March 2017, but there is a ready to use trace function with autograd capabilities now: torch.trace 196.
Just mentioning for everyone that gets here using a search engine.