jax.numpy.cross

jax.numpy.cross(a, b, axisa=-1, axisb=-1, axisc=-1, axis=None)

返回两个（向量数组）的叉积。

LAX 后端实现 numpy.cross().

原始文档字符串如下。

在 \(R^3\) 中，a 和 b 的叉积是一个垂直于 a 和 b 的向量。如果 a 和 b 是向量数组，则向量默认由 a 和 b 的最后一个轴定义，并且这些轴可以具有维度 2 或 3。当 a 或 b 的维度为 2 时，输入向量的第三个分量假定为零，并据此计算叉积。在两个输入向量都具有维度 2 的情况下，将返回叉积的 z 分量。

参数：

• a (array_like) – 第一个向量分量。

• b (array_like) – 第二个向量分量。

• axisa (int, 可选) – 定义向量(s)的 a 轴。默认情况下，最后一个轴。

• axisb (int, 可选) – b 的轴，定义向量。默认为最后一个轴。

• axisc (int, 可选) – c 的轴，包含叉积向量。如果两个输入向量都为二维，则忽略此参数，因为返回值为标量。默认为最后一个轴。

• axis (int, 可选) – 如果定义，则为 a、b 和 c 的轴，定义向量和叉积。覆盖 axisa、axisb 和 axisc。

返回:

c – 向量叉积。

返回类型:

ndarray