I am looking to batch multiply two matrices of dimensions torch.Size([7, 265, 768]).
When I tried with torch.bmm(x,y) I got the error like this,
RuntimeError: invalid argument 2: wrong matrix size, batch1: 265x768, batch2: 265x768 at /Users/soumith/code/builder/wheel/pytorch-src/aten/src/TH/generic/THTensorMath.cpp:2312
Any idea what might be causing this?