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