Hi everyone,i am having some trouble with torch.einsum.Basically i am trying to mutliply two tensors in a certain way:
first = torch.rand(12,8192,2)

weights1 = torch.rand(12,8192,2)

torch.einsum('bix,iox->box',first,weights1)

But i get the following error:
einsum() operands do not broadcast with remapped shapes [original->remapped]: [12, 8192, 2]->[12, 1, 2, 8192] [12, 8192, 2]->[1, 8192, 2, 12]
Can someone explain to me what I am doing wrong?