torch.geqrf

torch.geqrf(input, *, out=None)

這是一個直接呼叫 LAPACK 的 geqrf 的底層函式。此函式返回 LAPACK 文件中 geqrf 定義的 namedtuple (a, tau)。

計算 input 的 QR 分解。Q 矩陣和 R 矩陣都儲存在同一輸出張量 a 中。R 的元素儲存在對角線及上方。隱式定義矩陣 Q 的基本反射器（或 Householder 向量）儲存在對角線下方。此函式的結果可以與 torch.linalg.householder_product() 一起使用以獲取 Q 矩陣，或者與使用 Q 矩陣隱式表示形式的 torch.ormqr() 一起使用，以實現高效的矩陣-矩陣乘法。

有關更多詳細資訊，請參閱 LAPACK 文件中關於 geqrf 的說明。

注意

另請參閱 torch.linalg.qr()（計算 Q 和 R 矩陣），以及帶有 driver="gels" 選項的 torch.linalg.lstsq()（一個可以使用 QR 分解求解矩陣方程的函式）。

引數

input (Tensor) – 輸入矩陣

關鍵字引數

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