Python矢量化的矩阵矢量运算

Question

我有一个形状为(2,2，N(的矩阵a和形状为(2，N

我想矢量化以下内容：

F = np.zeros(N)
for k in xrange(N):
F[k] = np.dot( A[:,:,k], V[:,k] ).sum()

可以用tensordot或任何其他numpy函数在没有显式循环的情况下实现这一点吗？

Answer 1

带np.einsum-

F = np.einsum('ijk,jk->k',A,V)

我们可以通过将optimize标志(检查文档(设置为True来进一步优化它。