Solved by KFrank in post #4 
              
              
                Hi Abhisek! 

Could you post a trimmed-down, runnable, fully-self-contained script 
that reproduces the error?  (Please have the script print out the version 
of pytorch you’re using, torch.__version__.) 
Please provide hard-coded (or random) sample data (with size and 
dimensionality as small as po…