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/2.1/generated/torch.geqrf.html