jax.numpy.outer#
- jax.numpy.outer(a, b, out=None)[source]#
Compute the outer product of two vectors.
LAX-backend implementation of
numpy.outer()
.Original docstring below.
Given two vectors,
a = [a0, a1, ..., aM]
andb = [b0, b1, ..., bN]
, the outer product [1] is:[[a0*b0 a0*b1 ... a0*bN ] [a1*b0 . [ ... . [aM*b0 aM*bN ]]
- Parameters:
a ((M,) array_like) – First input vector. Input is flattened if not already 1-dimensional.
b ((N,) array_like) – Second input vector. Input is flattened if not already 1-dimensional.
- Returns:
out –
out[i, j] = a[i] * b[j]
- Return type:
(M, N) ndarray
References
- Parameters:
out (
None
) –