Find centralized, trusted content and collaborate around the technologies you use most.
Teams
Q&A for work
Connect and share knowledge within a single location that is structured and easy to search.
我想有效地计算大小为 nx3 和 mx3 的两个矩阵 A 和 B 的行的所有成对叉积。并且理想情况下希望以 einsum 表示法实现这一点。
即输出矩阵C,将是(n X mx 3),
在哪里
C[0][0] = 交叉(n[0],m[0])
C[0][1] = 交叉(n[0],m[1])
...
C[1][0] = 交叉(n[1],m[0])
由于我采用的方法,使用 for 循环不是一种选择。
任何帮助将非常感激。
看起来像交叉广播领先的维度。
np.cross(A[:, None,:], B[None, :,:])