Solved by ptrblck in post #6 
              
              
                This tutorial explains how to write custom autograd.Fucntions. 

If you are “just using” built-in PyTorch methods, you are most likely not interested in testing utilities. 
Yes, since PyTorch is an open source framework with a lot of active developers, things can accidentally break, which is why tes…