mindspore.ops.kron
- mindspore.ops.kron(input, other)[源代码]
计算 input 和 other 的Kronecker积。
如果 input 的shape为
, other 的shape为 ,则计算结果的shape为 。其中,对于所有的 0 ≤ t ≤ n,都有
。说明
支持实数和复数类型的输入。
- 参数:
input (Tensor) - 输入tensor。
other (Tensor) - 输入tensor 。
- 返回:
Tensor
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> input = mindspore.tensor([[0., 1., 2.], [3., 4., 5.]]) >>> other = mindspore.tensor([[-1., -2., -3.], [-4., -6., -8.]]) >>> output = mindspore.ops.kron(input, other) >>> print(output) [[ 0. 0. 0. -1. -2. -3. -2. -4. -6.] [ 0. 0. 0. -4. -6. -8. -8. -12. -16.] [ -3. -6. -9. -4. -8. -12. -5. -10. -15.] [-12. -18. -24. -16. -24. -32. -20. -30. -40.]]