Solved by KFrank in post #4 
              
              
                Hi @ptrblck! 

I can confirm that the issue has been fixed in the current nightly (that 
is, the cuda version also throws the shape-mismatch error): 
>>> torch.__version__
'1.10.0.dev20210830'
>>> torch.nn.Linear (3, 3).cuda() (torch.randn (5, 10).cuda())
Traceback (most recent call last):
  File "<s…