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!