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

ndarray