jax.numpy.kron
Warning
This page was created from a pull request (#9655).
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