【发布时间】:2014-05-11 11:48:49
【问题描述】:
This thread成功解答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]]
以便我可以在程序本身内对其执行进一步的操作。
我该怎么做?
【问题讨论】:
-
克罗内克张量积可以使用numpy.kron 计算,如果您正在搜索的话。如果不是,您能否给出您所追求的产品的定义?
-
@Phillip 看来您找到了解决方案,您可以发表评论作为答案...
-
哦,完美!这正是我所需要的!非常感谢。我希望答案是一个我可能忽略的简单答案。
-
很高兴听到。我已将我的评论重新发布为答案。