Computes the Kronecker (tensor) product between two mlx arrays. Inputs are
automatically cast to a common dtype and device before evaluation.
Arguments
- a, b
Objects coercible to mlx.
Value
An mlx array representing the Kronecker product.
Examples
A <- as_mlx(matrix(1:4, 2, 2))
B <- as_mlx(matrix(c(0, 5, 6, 7), 2, 2))
mlx_kron(A, B)
#> mlx array [4 x 4]
#> dtype: float32
#> device: gpu
#> values:
#> [,1] [,2] [,3] [,4]
#> [1,] 0 6 0 18
#> [2,] 5 7 15 21
#> [3,] 0 12 0 24
#> [4,] 10 14 20 28