Solved by ptrblck in post #2 
              
              
                Don’t use the .data attribute as it might yield unwanted side effects. 
While you will be able to manipulate the underlying data without raising an error, Autograd won’t be able to track these operations and you might run into a variety of issues later (we had quite a few of these issues already her…