Функция torch.geqrf в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.geqrf— это функция в PyTorch, используемая для вычисления QR-разложения матрицы, возвращающая матрицы Q и R.

Определение функции

torch.geqrf(input)

Возвращает:

  • a — ортогональная матрица Q
  • tau — коэффициент, необходимый для получения ортогональной матрицы

Пример использования

Пример

import torch

# Создать матрицу
A = torch.randn(3, 3)

# QR-разложение
a, tau = torch.geqrf(A)
print("a (Q):", a)
print("tau:", tau)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения