Python 二维和三维阵列之间的点积

假设我有两个数组

V

和

Q

，其中

V

是

（I，j，j）

，而

Q

是

（j，j）

。现在我想计算

Q

与

V

的每一行的点积，并将结果保存为

（I，j，j）

大小的矩阵。使用for循环很容易做到这一点，只需像这样在

i

上迭代即可

import numpy as np

v = np.random.normal(size=(100, 5, 5))
q = np.random.normal(size=(5, 5))
output = np.zeros_like(v)

for i in range(v.shape[0]):
    output[i] = q.dot(v[i])

然而，这对于我的需要来说太慢了，我猜有一种方法可以使用

einsum

或

tensordot

来对这个操作进行矢量化，但我还没有找到它。有人能给我指一下正确的方向吗？谢谢

您当然可以使用，但以后需要交换轴，就像这样-

out = np.tensordot(v,q,axes=(1,1)).swapaxes(1,2)

out = np.einsum('ijk,lj->ilk',v,q)

有了它，就更直截了当了，就像这样-

out = np.tensordot(v,q,axes=(1,1)).swapaxes(1,2)

out = np.einsum('ijk,lj->ilk',v,q)