Fonction torch.linalg.lstsq de PyTorch

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

torch.linalg.lstsqCeci est une fonction du module d'algèbre linéaire de PyTorch utilisée pour résoudre des problèmes de moindres carrés linéaires. Elle résout argmin_x ||AX - B||_F.

Définition de fonction

torch.linalg.lstsq(A, B, rcond=None, driver=None)

Parameters:

  • A(Tensor) : Matrice des coefficients.
  • B(Tensor) : Matrice ou vecteur du membre de droite.
  • rcond(float, optionnel) : Nombre de condition utilisé pour tronquer les valeurs singulières.
  • driver(str, optionnel) : Sélection du solveur ; les choix sont 'gels', 'gelsd', 'gelsy'.

Valeur de retour:

  • torch.Tensor: Renvoie la solution des moindres carrés.

Exemple d'utilisation

Exemple

import torch

# Créer une matrice de coefficients et un vecteur du membre de droite
A = torch.tensor([[1.0, 1.0], [1.0, 2.0], [1.0, 3.0]], dtype=torch.float64)
B = torch.tensor([1.0, 2.0, 3.0], dtype=torch.float64)

# Résoudre les moindres carrés
X = torch.linalg.lstsq(A, B).solution

print("Matrice des coefficients A :")
print(A)
print("nVecteur du membre de droite B :")
print(B)
print("nSolution des moindres carrés X :")
print(X)

La sortie est :

系数矩阵 A:
tensor([[1., 1.],
        [1., 2.],
        [1., 3.]], dtype=torch.float64)
右侧向量 B:
tensor([1., 2., 3.], dtype=float64)
最小二乘解 X:
tensor([0.0000, 1.0000], dtype=float64)

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

Autres extensions