My question is the following:
I have 2 3D tensors, and with one dimension in common.
1st tensor dimensions: a x b x c
2nd tensor dimensions: a x d x e
I want to multiply these two tensors along the dimension they have in common (a), and such that their output will be a 4 d tensor (e.g. dimensions b x c x d x e). Pretty much multiplying each vector (with a elements) of tensor 1 by each vector (with a elements) of tensor 2.
I can’t think of an easy built-in way to accomplish this. I can think of iterating over the dimensions and performing torch.dot, but that does not look very good. Does any one know of a better way?
Thank you for reading.