python - 两个矩阵行的所有成对叉积

Question

我想有效地计算大小为 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 循环不是一种选择。

任何帮助将非常感激。

标签： pythonnumpymathjax

Answer 1

看起来像交叉广播领先的维度。

np.cross(A[:, None,:], B[None, :,:])