jax.numpy.outer#

jax.numpy.outer(a,b,out=None)[source]#

Compute the outer product of two arrays.

JAX implementation ofnumpy.outer().

Parameters:

• a (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray) – first input array, if not 1D it will be flattened.

• b (Array |ndarray |bool |number |bool |int |float |complex |TypedNdArray) – second input array, if not 1D it will be flattened.

• out (None) – unsupported by JAX.

Returns:

The outer product of the inputsa andb. Returned arraywill be of shape(a.size,b.size).

Return type:

Array

See also

• jax.numpy.inner(): compute the inner product of two arrays.

• jax.numpy.einsum(): Einstein summation.

Examples

>>>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)