I would like to do a “batched” tensordot. For instance say I have ten 3d vectors packed into a tensor x with x.size() = (10,3).
And I have ten 2x3 matrices packed into a tensor A with A.size() = (10, 2, 3).
I would like to multiply the ith matrix with the ith vector and get the results packed into a tensor y with y.size() = (10, 2).
How to do this?