torch.geqrf
- torch.geqrf(input, *, out=None)
-
这是一个低级函数，直接调用 LAPACK 的 geqrf。此函数返回一个 namedtuple (a, tau)，具体定义请参见 LAPACK 文档中的 geqrf。
计算
input的QR分解。矩阵Q和R都存储在同一输出张量a中。R矩阵的元素位于对角线及其上方，而定义矩阵Q的基本反射向量（或Householder向量）则位于对角线下方。此函数的结果可以与torch.linalg.householder_product()一起使用以获得Q矩阵，或者与torch.ormqr()一起使用，该函数利用Q矩阵的隐式表示形式来实现高效的矩阵-矩阵乘法。有关 geqrf 的更多细节，请参阅 LAPACK 文档。
注意
另见
torch.linalg.qr()，该函数计算 Q 和 R 矩阵。此外，带有driver="gels"选项的torch.linalg.lstsq()函数也可以使用 QR 分解来求解矩阵方程。