python - 列表中的 sympy 点积结果

Question

我有两个 sympy 矩阵，U 和 B:

>> U
<< Matrix([
   [1.0,   0,   0,   0],
   [  0, 1.0,   0,   0],
   [  0,   0, 1.0,   0],
   [  0,   0,   0, 1.0]])
>> B
<< Matrix([
   [sqrt(2)/2,  0.5*sqrt(2)*I,             0,          0],
   [        0,              0, 0.5*sqrt(2)*I,  sqrt(2)/2],
   [        0,              0, 0.5*sqrt(2)*I, -sqrt(2)/2],
   [sqrt(2)/2, -0.5*sqrt(2)*I,             0,          0]])

对它们应用点积会得到一个列表，而不是一个 4x4 矩阵:

>> U.dot(B)
<< [0.5*sqrt(2),
    0,
    0,
    0.5*sqrt(2),
    0.5*sqrt(2)*I,
    0,
    0,
    -0.5*sqrt(2)*I,
    0,
    0.5*sqrt(2)*I,
    0.5*sqrt(2)*I,
    0,
    0,
    0.5*sqrt(2),
    -0.5*sqrt(2),
    0]

相比之下，numpy 看起来像是在做正确的事情:

>> numpy.dot(sympy.matrix2numpy(U),sympy.matrix2numpy(B))
<< array([[0.5*sqrt(2), 0.5*sqrt(2)*I, 0, 0],
   [0, 0, 0.5*sqrt(2)*I, 0.5*sqrt(2)],
   [0, 0, 0.5*sqrt(2)*I, -0.5*sqrt(2)],
   [0.5*sqrt(2), -0.5*sqrt(2)*I, 0, 0]], dtype=object)

我做错了什么？这是预期的行为吗？

Answer 1

与NumPy不同，在SymPy中，dot表示向量的点积。它的设计是为了让你可以得到两个行或列向量的点积，而不必担心使用 .T，但也许这里的形状松懈有点多，因为它实际上是取点list(U) 和 list(B) 的乘积。

这里可能太松了。我开了一个SymPy issue为了这。在这种情况下，SymPy 引发异常更为正确。

正如@Wrzlprmft 正确指出的那样，SymPy 使用* 来乘以矩阵(如果您使用的是Python 3.5 或更高版本，则使用@)。