Hi Paula (and Alex)!



 paula_gomez_duran:

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].


Building on Alex’s suggestion, I believe that
X.prod (dim = 1).sum (dim = 1)

should do what you want.
Best.
K. Frank