Fonction torch.linalg.solve de PyTorch

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

torch.linalg.solveest une fonction du module d'algèbre linéaire de PyTorch utilisée pour résoudre des équations linéaires. Elle résout AX = B.

Définition de fonction

torch.linalg.solve(A, B, left=True, out=None)

Parameters:

  • A(Tensor) : Matrice des coefficients.
  • B(Tensor) : Matrice ou vecteur du membre de droite.
  • left(bool, optionnel) : Si True, résout AX = B ; sinon, résout XA = B. La valeur par défaut est True.
  • out(Tensor, optionnel) : Tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie la solution des équations linéaires.

Exemple d'utilisation

Exemple

import torch

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

# Résoudre AX = B
X = torch.linalg.solve(A, B)

print("Matrice des coefficients A :")
print(A)
print("nVecteur du membre de droite B :")
print(B)
print("nSolution X:")
print(X)
print("nVérification: A @ X =")
print(A @ X)

Le résultat de sortie est:

系数矩阵 A:
tensor([[1., 2.],
        [3., 4.]])
右侧向量 B:
tensor([ 5., 11.])
解 X:
tensor([1., 2.])
验证: A @ X =
tensor([ 5., 11.])

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

Autres extensions