多矩阵乘法

Question

在numpy中，我有N个3x3矩阵的数组。这将是我如何存储它们的示例（我正在提取内容）：

N = 10
matrices = np.ones((N, 3, 3))

我也有一个由3个向量组成的数组，这将是一个示例：

vectors = np.ones((N, 3))

我似乎无法弄清楚如何通过numpy将它们相乘，从而实现如下效果：

result_vectors = []
for matrix, vector in zip(matrices, vectors):
    result_vectors.append(matrix @ vector)

与result_vector的形状（在投射到阵列）是(N, 3)。但是，由于速度的原因，列表实现是不可能的。

我尝试了各种换位的np.dot，但最终结果没有得到正确的形状。

Answer 1

使用np.einsum

np.einsum('ijk,ik->ij',matrices,vectors)

脚步 ：

1）保持第一根轴对齐。

2）将输入数组中的最后一个轴相减。

3）剩余的轴（从的第二轴matrices）乘以元素方式。