Python numpy linalg.tensorsolve用法及代碼示例

本文簡要介紹 python 語言中 numpy.linalg.tensorsolve 的用法。

用法:

linalg.tensorsolve(a, b, axes=None)

求解 x 的張量方程 a x = b。

假設所有指數x與最右邊的指數一起在乘積中求和a，例如，tensordot(a, x, axes=b.ndim).

參數：

a： array_like

係數張量，形狀b.shape + Q.Q，一個元組，等於 sub-tensor 的形狀a由適當數量的最右邊的索引組成，並且必須是這樣的prod(Q) == prod(b.shape)(在什麽意義上a據說是‘square’)。

b： array_like

右手張量，可以是任何形狀。

axes： 整數元組，可選

a 中的軸在反轉之前向右重新排序。如果 None (默認)，則不進行重新排序。

返回：

x： ndarray，形狀 Q

拋出：

LinAlgError

如果 a 是單數或不是‘square’(在上述意義上)。

例子：

>>> a = np.eye(2*3*4)
>>> a.shape = (2*3, 4, 2, 3, 4)
>>> b = np.random.randn(2*3, 4)
>>> x = np.linalg.tensorsolve(a, b)
>>> x.shape
(2, 3, 4)
>>> np.allclose(np.tensordot(a, x, axes=3), b)
True