torch.geqrf
-
torch.geqrf(input, *, out=None) -
Это функция низкого уровня для прямого вызова LAPACK’s 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-разложения.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.geqrf.html