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