python - 用 numpy (einsum) 向量化多维矩阵乘积

Question

我想以矢量化方式（1000xmxm dot 1000xnxm --> 1000xnxm）在一系列 nxm 矩阵上使用 mxm 数组执行一系列旋转。 如果我的 nxm 矩阵是 1xm 向量，这个问题似乎有答案，但不幸的是它们不是。我想知道是否有一个我可以使用的 einsum，或者其他一些矩阵操作不必循环超过 1000 次。