Solved by albanD in post #2 
              
              
                Hi, 
Are you using a custom autograd.Function ? This error is because you try to use a custom autograd Function without implementing the backward.