torch.linalg.solve_ex

torch.linalg.solve_ex(A, B, *, left=True, check_errors=False, out=None)

此為 solve() 的一個版本，除非 check_errors= True，否則不會執行錯誤檢查。 它也會回傳 LAPACK 的 getrf 所回傳的 info 張量。

注意

當輸入位於 CUDA 裝置上時，此函式僅在 check_errors= True 時才會同步。

警告

此函式為「實驗性」功能，未來的 PyTorch 版本可能會有所變更。

參數

A (Tensor) – 形狀為 (*, n, n) 的張量，其中 * 是零或多個批次維度。

關鍵字參數

• left (bool, optional) – 是否求解系統 $AX=B$ 或 $XA = B$。預設值：True。

• check_errors (bool, optional) – 控制是否檢查 infos 的內容，如果非零則引發錯誤。預設值：False。

• out (tuple, optional) – 用於寫入輸出的兩個張量的元組。如果為 None 則忽略。預設值：None。

回傳值

一個名為 (result, info) 的具名元組。

範例

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