-1

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

任何帮助将非常感激。

4

1 回答 1

1

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

np.cross(A[:, None,:], B[None, :,:])
于 2021-04-22T04:40:14.487 回答