torch.linalg.lu_factor_ex

torch.linalg.lu_factor_ex(A, *, pivot=True, check_errors=False, out=None)

這是 lu_factor() 的一個版本，除非 check_errors= True，否則不執行錯誤檢查。它還返回 LAPACK 的 getrf 函式返回的 info 張量。

注意

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

警告

此函式是“實驗性”的，可能會在未來的 PyTorch 版本中更改。

引數

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

關鍵字引數

• pivot (布林型, 可選) – 是否計算帶部分主元的 LU 分解或常規 LU 分解。pivot= False 在 CPU 上不受支援。預設值：True。

• check_errors (布林型, 可選) – 控制是否檢查 infos 的內容並在其非零時引發錯誤。預設值：False。

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

返回

一個命名元組 (LU, pivots, info)。