Incorrect multiplication order for einsum when aggregating adjacent node information. (vw -> wv) about mtgnn

[nnzhan]

Hi thanks for your comment, I think you are right. After further investigation, I will correct it.

[ntubiolin]

OK, thanks!

[lorrp1]

@ntubiolin @nnzhan hello, im getting this error:

return _VF.einsum(equation, operands)
RuntimeError: size of dimension does not match previous size, operand 1, dim 0

in

 MTGNN-master/layer.py", line 14, in forward
    x = torch.einsum('ncvl,vw->ncwl',(x,A))

have you got any idea on how i could solve it?

[nnzhan]

@lorrp1 Can you print out the shape of X and A?

[zhp66]

@ntubiolin @nnzhan Hello，I had the same problem
return torch._C._VariableFunctions.einsum(equation, operands) RuntimeError: size of dimension does not match previous size, operand 1, dim 1
in
MTGNN-master/layer.py", line 14, in forward x = torch.einsum('ncvl,vw->ncwl',(x,A))
and I printed out the shapes of X and A
torch.Size([64, 32, 206, 13]) torch.Size([207, 207])
I'm using the METR-LA dataset