Fonction torch.ormqr de PyTorch

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

torch.ormqrIl s'agit d'une fonction de PyTorch qui utilise la décomposition QR pour la multiplication matricielle. Elle calcule Q @ input ou input @ Q, où Q est une matrice orthogonale obtenue à partir de la décomposition QR.

Définition de fonction

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

Parameters:

  • input(Tensor): matrice d'entrée.
  • tau(Tensor) : coefficients des réflecteurs de Householder.
  • left(bool, facultatif) : Si True, calcule Q @ input ; sinon calcule input @ Q. Valeur par défaut : True.
  • transpose(bool, facultatif) : Indique s'il faut transposer Q. Valeur par défaut : False.
  • out(Tensor, optionnel) : Tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie le résultat de la multiplication matricielle.

Exemple d'utilisation

Exemple

import torch

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

# Décomposition QR
Q, R = torch.linalg.qr(A)

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

print("Shape of matrix B:", B.shape)
print("Forme du résultat :", result.shape)

Le résultat de sortie est:

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

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

Autres extensions