jax.numpy.kron

jax.numpy.kron(a, b)

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