jax.numpy.kron#
- jax.numpy.kron(a, b)[source]#
Kronecker product of two arrays.
LAX-backend implementation of
numpy.kron()
.Original docstring below.
Computes the Kronecker product, a composite array made of blocks of the second array scaled by the first.
- Parameters:
a (array_like) –
b (array_like) –
- Returns:
out
- Return type:
ndarray