Python 3.x 将许多矩阵和向量相乘

我正试图将以下各项相乘：

一批矩阵

nxmxd

一批向量

nxdx1

要获得结果：

nxmx1

就像我在

md

dx1

上做

N

点积一样

我似乎在PyTorch中找不到正确的函数

---

torch.bmm

据我所知，它只适用于一批向量和一个矩阵。如果我必须使用torch.einsum，那就这样吧，但我宁愿不要

使用

einsum

非常直观：

torch.einsum('ijk, ikl->ijl', mats, vecs)

但你的行动只是：

mats @ vecs