評價此頁

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 分解求解矩陣方程的函式。

引數

input (Tensor) – 輸入矩陣

關鍵字引數

out (tuple, optional) – 輸出元組 (Tensor, Tensor)。如果為 None 則忽略。預設為 None