mindspore.numpy.kron

查看源文件
mindspore.numpy.kron(a, b)[源代码]

两个数组的Kronecker积。

计算Kronecker积,即先将第二个数组分块,然后使用第一个数组缩放后组成的复合数组。

说明

不支持bool值。

参数:
  • a (Union[int, float, list, tuple, Tensor]) - 输入值。

  • b (Union[int, float, list, tuple, Tensor]) - 输入值。

返回:

Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> output = np.kron([1,10,100], [5,6,7])
>>> print(output)
[  5   6   7  50  60  70 500 600 700]
>>> output = np.kron([5,6,7], [1,10,100])
>>> print(output)
[  5  50 500   6  60 600   7  70 700]
>>> output = np.kron(np.eye(2), np.ones((2,2)))
>>> print(output)
[[1. 1. 0. 0.]
[1. 1. 0. 0.]
[0. 0. 1. 1.]
[0. 0. 1. 1.]]