torch.geqrf#
- torch.geqrf(input, *, out=None)#
這是一個直接呼叫 LAPACK 的 geqrf 的底層函式。此函式返回一個命名元組 (a, tau),其定義如 LAPACK geqrf 文件 所述。
計算
input的 QR 分解。Q 和 R 矩陣都儲存在同一個輸出張量 a 中。R 的元素儲存在對角線及其上方。隱式定義矩陣 Q 的初等反射(或 Householder 向量)儲存在對角線下方。此函式的結果可以與torch.linalg.householder_product()一起使用以獲得 Q 矩陣,或者與torch.ormqr()一起使用,它使用 Q 矩陣的隱式表示,以實現高效的矩陣-矩陣乘法。有關更多詳細資訊,請參閱 LAPACK geqrf 文件。
注意
另請參閱
torch.linalg.qr(),它計算 Q 和 R 矩陣,以及帶driver="gels"選項的torch.linalg.lstsq(),這是一個可以使用 QR 分解求解矩陣方程的函式。