I have a batch = 256 of pair of vectors of dimension 32. The shape is so [256, 2, 32].
What I would want would be to perform the dot product between each of the pair of vectors of the batch, so that I end up with a tensor of shape [256, 32].
How should I do it ?
Thank you in advance!