Fonction torch.solve de PyTorch

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

torch.solveC'est une fonction de PyTorch pour résoudre des équations linéaires. Elle résout AX = B, où A est la matrice des coefficients et B est la matrice ou le vecteur du membre de droite.

Définition de fonction

torch.solve(B, A, out=None)

Parameters:

  • B(Tensor) : Matrice ou vecteur du membre de droite.
  • A(Tensor) : Matrice des coefficients.
  • out(tuple, facultatif) : Tuple de sortie.

Valeur de retour:

  • tuple: Renvoie un tuple (X, LU), où X est la solution et LU est la décomposition.

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.solve(B, A)

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