jax.numpy.kron¶
-
jax.numpy.kron(a, b)[source]¶ Kronecker product of two arrays.
LAX-backend implementation of
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