jax.numpy.outer

jax.numpy.outer(a, b, out=None)

计算两个数组的外积。

JAX 实现的 numpy.outer()。

参数:

• a (Array | ndarray | bool | number | bool | int | float | complex) – 第一个输入数组，如果不是 1D 数组，则会被展平。

• b (Array | ndarray | bool | number | bool | int | float | complex) – 第二个输入数组，如果不是 1D 数组，则会被展平。

• out (None) – JAX 不支持。

返回值:

输入 a 和 b 的外积。返回数组的形状为 (a.size, b.size)。

返回类型:

Array

参见

• jax.numpy.inner(): 计算两个数组的内积。

• jax.numpy.einsum(): 爱因斯坦求和。

示例

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