mlx.core.kron

kron(a: array, b: array, *, stream: None | Stream | Device = None) → array

计算两个数组 a 和 b 的克罗内克积。

参数:

• a (array) – 第一个输入数组。

• b (array) – 第二个输入数组。

• stream (Union[None, Stream, Device], 可选) – 可选的执行流或设备。默认为 None。

返回:

a 和 b 的克罗内克积。

返回类型:

array

示例

>>> a = mx.array([[1, 2], [3, 4]])
>>> b = mx.array([[0, 5], [6, 7]])
>>> result = mx.kron(a, b)
>>> print(result)
array([[0, 5, 0, 10],
       [6, 7, 12, 14],
       [0, 15, 0, 20],
       [18, 21, 24, 28]], dtype=int32)