jax.numpy.cross

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

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

LAX-backend 实现的 numpy.cross()。

原始文档字符串如下。

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

参数:

• a (array_like) – 第一个向量(s)的组成部分。

• b (array_like) – 第二个向量(s)的组成部分。

• axisa (int, optional) – 定义向量（或多个向量）的 a 的轴。默认情况下，为最后一个轴。

• axisb (int, optional) – 定义向量（或多个向量）的 b 轴。默认情况下，为最后一个轴。

• axisc (int, optional) – 包含叉积向量(s)的 c 轴。如果两个输入向量的维度都是2，则忽略此参数，因为返回值是标量。默认情况下，使用最后一个轴。

• axis (int, optional) – 如果定义了，a、b 和 c 的轴，用于定义向量和叉积。覆盖 axisa、axisb 和 axisc。

返回:

c – 向量叉积。

返回类型:

ndarray