python - numpy中的多个矩阵产品

Question

我有两个 numpy 数组A，分别B是形状(k, m, n)和形状(k, n, l)。我有兴趣获得以下 numpy 数组：

[A[0]@B[0], A[1]@B[1], ..., A[k-1]@B[k-1]]

显然，我可以使用 for 循环来做到这一点，但我想知道是否有更“numpy”的方式来做这件事，它可能会表现得更好。

谢谢。

标签： pythonnumpymatrix-multiplication