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 矩阵，以及torch.linalg.lstsq()（带driver="gels"选项），这是一个可以使用 QR 分解来求解矩阵方程的函数。