jax.numpy.linalg.cross#

jax.numpy.linalg.cross(x1, x2, /, *, axis=-1)[原始碼]#

計算兩個 3D 向量的外積

numpy.linalg.cross() 的 JAX 實作

參數:

  • x1 (ArrayLike) – N 維陣列，其中 x1.shape[axis] == 3

  • x2 (ArrayLike) – N 維陣列，其中 x2.shape[axis] == 3，且其他軸與 x1 的廣播相容。

  • axis – 沿哪個軸取外積（預設值：-1）。

傳回值:

包含外積結果的陣列

另請參閱

jax.numpy.cross()：更彈性的外積 API。

範例

顯示 \(\hat{x} \times \hat{y} = \hat{z}\)

>>> x = jnp.array([1., 0., 0.])
>>> y = jnp.array([0., 1., 0.])
>>> jnp.linalg.cross(x, y)
Array([0., 0., 1.], dtype=float32)

透過廣播，\(\hat{x}\) 與所有三個標準單位向量的外積

>>> xyz = jnp.eye(3)
>>> jnp.linalg.cross(x, xyz, axis=-1)
Array([[ 0.,  0.,  0.],
       [ 0.,  0.,  1.],
       [ 0., -1.,  0.]], dtype=float32)