标签： python numpy
我正在尝试使用np.einsum对向量（v）和两个矩阵（B）进行以下求和
A [i，j] =（从k = 1到N的总和）v [k] * B [k，i] * B [k，j]
A = np.einsum('k,ki,kj->ij',v,B,B)
我希望对索引k的求和将导致一个i x j矩阵，但是不确定是执行求和还是乘积