numpy中Kronecker积的双循环矢量化和整形

Question

我有两个矩阵

import numpy as np
n = 10
a = 2*np.ones((n,n,3))
b = 3*np.ones((n,n,3))

我想以提醒Kronecker产品的方式将它们相乘，然后进行总结

s = 0
for i in range(n):
    for j in range(n):
        s +=  a*b[i,j]

是否存在将其向量化为numpy的方法？

Answer 1

您的代码可以重写为：

因此，这应该起作用：

s = a * np.sum(np.sum(b,axis=1),axis=0)

Answer 2

也许这可以用np.einsum()更加优雅地写：

np.einsum()