I’m tying to calculate the matrix-tensor product between a bxd matrix (a batch of d-dimensional vectors) and a bxdxn tensor (a batch of dxn matrices), so as the product is a bxn matrix, so as each row of the result is the product (a vector) between the corresponding vector of the matrix and the matrix of the tensor.
For instance:
import torch
X = torch.randn(32, 512)
Y = torch.randn(32, 512, 128)
# Z: needs to be of torch.Size([32, 128])

Thanks!