torch.geqrf

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

Это низкоуровневая функция для прямого вызова LAPACK geqrf. Эта функция возвращает именованный кортеж (a, tau), определённый в документации LAPACK по geqrf .

Вычисляет QR-разложение input. Матрицы Q и R хранятся в одном выходном тензоре a. Элементы R хранятся на диагонали и выше неё. Элементарные отражения (или векторы Хаусхолдера), неявно задающие матрицу Q, хранятся ниже диагонали. Результаты этой функции можно использовать вместе с torch.linalg.householder_product() для получения матрицы Q или с torch.ormqr(), которая использует неявное представление матрицы Q для эффективного умножения матриц.

Дополнительные сведения см. в документации LAPACK по geqrf.

Примечание

См. также torch.linalg.qr(), которая вычисляет матрицы Q и R, и torch.linalg.lstsq() с параметром driver="gels" для функции, решающей матричные уравнения с помощью QR-разложения.

Параметры:

input (Tensor) – входная матрица

Именованные аргументы:

out (tuple, необязательный) – выходной кортеж (Tensor, Tensor). Игнорируется, если None. Значение по умолчанию: None.