如何在Python中生成两个矩阵的可用张量积

Question

这个讨论帖成功地回答了如何在Python中使用numpy来计算两个矩阵的张量积。不过，我希望输出的结果能以合适的矩阵格式呈现，这样我就可以在不手动编辑的情况下进行进一步的计算。

比如，

I=matrix([[1,0],[0,1]])
print np.tensordot(I, I, axes=0)

会得到这样的输出：

[[[[1 0]
   [0 1]]

  [[0 0]
   [0 0]]]


 [[[0 0]
   [0 0]]

  [[1 0]
   [0 1]]]]

我希望它能呈现成：

[[1,0,0,0],[0,1,0,0],[0,0,1,0],[0,0,0,1]]

这样我就可以在程序内部进行进一步的操作了。

我该怎么做呢？

Answer 1

克罗内克积可以通过 numpy.kron 这个函数来计算。