Python PyTorch inv_ex用法及代碼示例

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

用法:

torch.linalg.inv_ex(A, *, check_errors=False, out=None)

參數：

• A(Tensor) -形狀為 (*, n, n) 的張量，其中 * 是零個或多個由方陣組成的批量維度。

• check_errors(bool,可選的) -控製是否檢查 info 的內容。默認值：False。

關鍵字參數：

out(tuple,可選的) -將輸出寫入的兩個張量的元組。如果 None 則忽略。默認值：None。

如果方陣可逆，則計算它的逆矩陣。

返回一個命名元組 (inverse, info) 。 inverse 包含反轉 A 的結果，info 存儲 LAPACK 錯誤代碼。

如果A不是可逆矩陣，或者它是一批矩陣，其中一個或多個不是可逆矩陣，則info為相應的矩陣存儲一個正整數。正整數表示輸入矩陣的 LU 分解的對角元素正好為零。 info補零表示反演成功。如果check_errors=True 和info 包含正整數，則拋出RuntimeError。

支持 float、double、cfloat 和 cdouble dtypes 的輸入。還支持批量矩陣，如果 A 是批量矩陣，則輸出具有相同的批量維度。

注意

如果 A 在 CUDA 設備上，則此函數可以將該設備與 CPU 同步。

警告

此函數是“experimental”，它可能會在未來的PyTorch 版本中更改。

例子：

>>> A = torch.randn(3, 3)
>>> Ainv, info = torch.linalg.inv_ex(A)
>>> torch.dist(torch.linalg.inv(A), Ainv)
tensor(0.)
>>> info
tensor(0, dtype=torch.int32)