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

Question

This thread成功地回答了如何使用numpy来计算Python中两个矩阵的张量积。但是，我希望输出采用适当的矩阵格式，以便可以对其进行进一步的计算，而无需手动编辑。

例如，

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计算Kronecker张量积。