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 矩阵，或者与使用 Q 矩阵隐式表示的 torch.ormqr() 一起使用，以实现高效的矩阵-矩阵乘法。

有关 geqrf 的详细信息，请参阅 LAPACK 文档。

注意

还请参阅 torch.linalg.qr() ，它计算 Q 和 R 矩阵，以及具有 driver="gels" 选项的 torch.linalg.lstsq() ，该选项是一个可以使用 QR 分解求解矩阵方程的函数。

参数:

输入（张量）- 输入矩阵

关键字参数:

out（元组，可选）- 输出元组（Tensor，Tensor）。如果为 None 则忽略。默认：None。