jax.numpy.linalg.outer#

jax.numpy.linalg.outer(x1,x2,/)[source]#

Compute the outer product of two 1-dimensional arrays.

JAX implementation ofnumpy.linalg.outer().

Parameters:

• x1 (ArrayLike) – array

• x2 (ArrayLike) – array

Returns:

array containing the outer product ofx1 andx2

Return type:

Array

See also

jax.numpy.outer(): similar function in the mainjax.numpy module.

Examples

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