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

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

用法:

linalg.tensorinv(a, ind=2)

計算 N 維數組的‘inverse’。

結果是倒數a相對於tensordot操作tensordot(a, b, ind)， 我。即，高達浮點精度，tensordot(tensorinv(a), a, ind)是用於tensordot 運算的“identity” 張量。

參數：

a： array_like

張量到‘invert’。它的形狀必須是‘square’，即。例如，prod(a.shape[:ind]) == prod(a.shape[ind:])。

ind： 整數，可選

反和中涉及的第一個索引的數量。必須為正整數，默認為 2。

返回：

b： ndarray

a的張量點逆，形狀a.shape[ind:] + a.shape[:ind].

拋出：

LinAlgError

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

例子：

>>> a = np.eye(4*6)
>>> a.shape = (4, 6, 8, 3)
>>> ainv = np.linalg.tensorinv(a, ind=2)
>>> ainv.shape
(8, 3, 4, 6)
>>> b = np.random.randn(4, 6)
>>> np.allclose(np.tensordot(ainv, b), np.linalg.tensorsolve(a, b))
True

>>> a = np.eye(4*6)
>>> a.shape = (24, 8, 3)
>>> ainv = np.linalg.tensorinv(a, ind=1)
>>> ainv.shape
(8, 3, 24)
>>> b = np.random.randn(24)
>>> np.allclose(np.tensordot(ainv, b, 1), np.linalg.tensorsolve(a, b))
True