问题描述
我想有效地计算大小为 nx3 和 mx3 的两个矩阵 A 和 B 的行的所有成对叉积。并且理想情况下希望以 einsum 表示法实现这一点。
即输出矩阵 C,将是 (n X m x 3),
哪里
C[0][0] = cross(n[0],m[0])
C[0][1] = cross(n[0],m[1])
...
C[1][0] = cross(n[1],m[0])
...
由于我采用的方法,不能使用 for 循环。
任何帮助将不胜感激。
解决方法
看起来像交叉广播的主要维度。
np.cross(A[:,None,:],B[None,:,:])