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

Parameters

输入 (张量) – 输入矩阵

Keyword Arguments

输出 (元组, 可选) – 输出元组 (Tensor, Tensor)。如果为 None，则忽略。默认值：None。