多个矩阵的张量积

Question

我想用 sympy 的 TensorProduct 来计算两个以上矩阵的张量积。

下面这个例子可以正常运行：

from sympy.physics.quantum import TensorProduct
from sympy import Matrix

m1 = Matrix([[1,2],[3,4]])
m2 = Matrix([[1,0],[0,1]])
m3 = Matrix([[1,1],[1,1]])

prod = TensorProduct(m1, m2, m3)

我想知道是否可以像这样做：

matrices = [m1, m2, m3]
prod = TensorProduct(matrices)

我尝试把 matrices 定义成一个数组和一个元组，但都不行。没有找到其他可以做到这一点的函数。

Answer 1

你只需要把 matrices 拆开，就像这样：

matrices = [m1, m2, m3]
prod = TensorProduct(*matrices)
prod