torch.linalg.lu_factor_ex

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

這是 lu_factor() 的一個版本，除非 check_errors= True，否則不會執行錯誤檢查。它也會回傳 LAPACK’s getrf 回傳的 info 張量。

注意

當輸入在 CUDA 裝置上時，只有在 check_errors= True 時，此函數才會同步。

警告

此函數是「實驗性」的，並且可能在未來的 PyTorch 版本中發生變更。

參數

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

關鍵字參數

• pivot (bool, optional) – 是否使用部分軸點 (partial pivoting) 計算 LU 分解，或使用常規 LU 分解。CPU 不支援 pivot= False。預設值：True。

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

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

回傳

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