jax.numpy.linalg.outer#

jax.numpy.linalg.outer(x1, x2, /)[源代码]#

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

numpy.linalg.outer() 的 JAX 实现。

参数:

  • x1 (类数组) – 数组

  • x2 (类数组) – 数组

返回:

包含 x1 和 x2 外积的数组

返回类型:

数组

另请参阅

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)