Fonction torch.geqrf de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.geqrfC'est une fonction dans PyTorch utilisée pour calculer la décomposition QR d'une matrice, en renvoyant les matrices Q et R.

Définition de fonction

torch.geqrf(input)

Retourne :

  • a - Matrice orthogonale Q
  • tau - Facteur requis pour produire la matrice orthogonale

Exemple d'utilisation

Exemple

import torch

# Créer une matrice
A = torch.randn(3, 3)

# Décomposition QR
a, tau = torch.geqrf(A)
print("a (Q):", a)
print("tau:", tau)

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions