Solved by ptrblck in post #6 
              
              
                If you need to call a numpy method e.g. since there is no corresponding method in PyTorch, your workflow is correct. The no_grad() wrapper wouldn’t change anything, which was the initial question, since Autograd won’t track any numpy operations. 
You can still use the processing in the forward metho…