torch.geqrf

torch.geqrf(input, *, out=None)

这是一个直接调用 LAPACK 的 geqrf 的底层函数。此函数返回 LAPACK 文档中 geqrf 定义的 namedtuple (a, tau)。

计算 input 的 QR 分解。Q 矩阵和 R 矩阵都存储在同一输出张量 a 中。R 的元素存储在对角线及上方。隐式定义矩阵 Q 的基本反射器（或 Householder 向量）存储在对角线下方。此函数的结果可以与 torch.linalg.householder_product() 一起使用以获取 Q 矩阵，或者与使用 Q 矩阵隐式表示形式的 torch.ormqr() 一起使用，以实现高效的矩阵-矩阵乘法。

有关更多详细信息，请参阅 LAPACK 文档中关于 geqrf 的说明。

注意

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

参数

input (Tensor) – 输入矩阵

关键字参数

out (tuple, optional) – 输出元组 (Tensor, Tensor)。如果为 None 则忽略。默认值：None。