torch.geqrf¶
- torch.geqrf(input, *, out=None)¶
這是用於直接呼叫 LAPACK 的 geqrf 的低階函式。此函式會傳回一個 namedtuple (a, tau)，如 geqrf 的 LAPACK 文件 中所定義。
計算
input的 QR 分解。 Q 和 R 矩陣都儲存在同一個輸出張量 a 中。R 的元素儲存在對角線及其上方。隱式定義矩陣 Q 的基本反射器（或 Householder 向量）儲存在對角線下方。此函數的結果可以與torch.linalg.householder_product()一起使用，以獲得 Q 矩陣；或者與torch.ormqr()一起使用，後者使用 Q 矩陣的隱式表示，以實現高效的矩陣-矩陣乘法。有關更多詳細信息，請參閱 LAPACK documentation for geqrf。
注意
另請參閱
torch.linalg.qr()，它計算 Q 和 R 矩陣；以及torch.linalg.lstsq()，使用driver="gels"選項，這是一個可以使用 QR 分解求解矩陣方程的函數。
