Hi!
I want to code this matrix multiplication:

B=16, h=32, w=56, b*b=64, n=25
my code:
torch.mul(torch.rand([16, 32, 56, 64]),torch.rand([16, 32, 56, 64,25]))

but I encountered in with this error:
The size of tensor a (64) must match the size of tensor b (25) at non-singleton dimension 4

Would you mind helping me to fix this problem?