两个矩阵的行的所有成对叉积
All pairwise cross products of the rows of two matrices
我想有效地计算大小为 nx3 和 mx3 的两个矩阵 A 和 B 的行的所有成对叉积。并且理想情况下希望以 einsum 表示法实现这一点。
即输出矩阵 C 为 (n X m x 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, :,:])
我想有效地计算大小为 nx3 和 mx3 的两个矩阵 A 和 B 的行的所有成对叉积。并且理想情况下希望以 einsum 表示法实现这一点。
即输出矩阵 C 为 (n X m x 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, :,:])