首页 > 解决方案 > 用 numpy (einsum) 向量化多维矩阵乘积

问题描述

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

标签: pythonnumpynumpy-einsum

解决方案


推荐阅读