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

Question

这个线程成功地回答了如何使用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张量积.