Python：将向量列表与矩阵列表相乘作为单个矩阵运算

您可以np.einsum像这样使用-

np.einsum('ij,ikj->ik',a,b)


样品运行-

In [42]: M,N = 3,4
In [43]: a = np.random.rand(100,N)
In [44]: b = np.random.rand(100,M,N)
In [45]: np.einsum('ij,ikj->ik',a,b).shape
Out[45]: (100, 3)


您也可以使用np.matmul或@运算符（Python 3.x），尽管它似乎比einsum-慢一点-

np.matmul(a[:,None],b.swapaxes(1,2))[:,0]