Solved by Nikronic in post #2 
              
              
                Hi, 
Could please sure a code that can reproduces this error? 
The problem is torch.dot() only supports 1D tensors. So, the error should be about it. The alternatives function for 2D and 3D (batch-wise) is torch.mm and torch.bmm respectively. 
Here is the code I used to produce your case (although c…