Solved by ptrblck in post #2 
              
              
                This line of code 

unsqueezes the tensor twice, which will most likely create the issue. 
Remove one None and rerun your code.