tensorprod

tensorprod(*args)

Return the Kronecker tensor product of arrays. It extends the functionality of the (jax.)numpy.kron() function.

Parameters:

*args – Input arrays to compute the Kronecker tensor product.

Return type:

ndarray | Array

Returns:

Kronecker tensor product of the input arrays