python - 带有 numpy 的矢量化内积

Question

我需要为存储在 numpy 数组中的向量计算许多内积。

也就是说，用数学术语来说，

V _i = Σ _k A _i,k B _i,k

和

C _i,j = Σ _k A _i,k D _i,j,k

我当然可以使用循环，但我想知道我是否可以使用更高级别的运算符，dot或者matmul以我没有想到的巧妙方式。困扰我的是它np.cross确实接受要执行的向量数组，但dot不做inner我想做的事。

标签： pythonnumpylinear-algebra

Answer 1

您的数学公式已经为您提供了使用np.einsum. 它很简单：

V = np.einsum('ij,ik->i', A, B)

转换为V[i] = sum(A[i][k]*B[i][k]).

和

C = np.einsum('ik,ijk->ij', A, D)

IE C[i][j] = sum(A[i][k]*D[i][j][k]

您可以在其他问题上阅读有关einsum操作员的更多信息。