jax.numpy.linalg.outer#

jax.numpy.linalg.outer(x1, x2, /)[原始碼]#

計算兩個 1 維陣列的外積。

JAX 實作的 numpy.linalg.outer()。

參數：

  • x1 (ArrayLike) – 陣列

  • x2 (ArrayLike) – 陣列

傳回：

包含 x1 和 x2 外積的陣列

傳回類型：

Array

另請參閱

jax.numpy.outer()：在主要 jax.numpy 模組中的類似函數。

範例

>>> x1 = jnp.array([1, 2, 3])
>>> x2 = jnp.array([4, 5, 6])
>>> jnp.linalg.outer(x1, x2)
Array([[ 4,  5,  6],
       [ 8, 10, 12],
       [12, 15, 18]], dtype=int32)