Función torch.geqrf de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.geqrfEs la función en PyTorch utilizada para calcular la descomposición QR de una matriz, devolviendo las matrices Q y R.

Definición de la función

torch.geqrf(input)

Devuelve:

  • a - matriz ortogonal Q
  • tau - factor necesario para generar la matriz ortogonal

Ejemplo de uso

Ejemplo

import torch

# Crear la matriz
A = torch.randn(3, 3)

# Descomposición QR
a, tau = torch.geqrf(A)
print("a (Q):", a)
print("tau:", tau)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones