Función torch.ormqr de PyTorch

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

torch.ormqrEs una función de PyTorch que utiliza la descomposición QR para la multiplicación de matrices. Calcula Q @ input o input @ Q, donde Q es una matriz ortogonal obtenida de la descomposición QR.

Definición de la función

torch.ormqr(input, tau, left=True, transpose=False, out=None)

Parámetros:

  • input(Tensor): La matriz de entrada.
  • tau(Tensor): Coeficientes del reflector de Householder.
  • left(bool, opcional): Si es True, calcula Q @ input; de lo contrario, calcula input @ Q. El valor predeterminado es True.
  • transpose(bool, opcional): Si se transpone Q. El valor predeterminado es False.
  • out(Tensor, opcional): El tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve el resultado de la multiplicación de matrices.

Ejemplo de uso

Ejemplo

import torch

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

# Descomposición QR
Q, R = torch.linalg.qr(A)

# Calcular Q @ B
result = torch.ormqr(B, torch.zeros(3))

print("Forma de la matriz B:", B.shape)
print("Forma del resultado:", result.shape)

El resultado de salida es:

矩阵 B 形状: torch.Size([3, 4])
结果形状: torch.Size([3, 4])

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

Otras extensiones