Hi Capangas!
einsum() is a fine solution – and a good general-purpose tool.
In your specific case, you can use torch.bmm() with a couple of
well-placed unsqueeze()s. A good discussion is found in this thread:
Best.
K. Frank
Hi Capangas!
einsum() is a fine solution – and a good general-purpose tool.
In your specific case, you can use torch.bmm() with a couple of
well-placed unsqueeze()s. A good discussion is found in this thread:
Best.
K. Frank