jax.numpy.linalg.outer

jax.numpy.linalg.outer(x1, x2, /)

计算两个一维数组的外积。

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)