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

参数

input (张量) – 输入矩阵

关键字参数

out (元组, 可选) – (张量, 张量) 的输出元组。如果为 None 则忽略。默认值：None。