Fonction torch.triangular_solve de PyTorch

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

torch.triangular_solveIl s'agit d'une fonction dans PyTorch utilisée pour résoudre des équations linéaires triangulaires. Cette fonction résout AX = B, où A est une matrice triangulaire.

Définition de fonction

torch.triangular_solve(A, B, upper, transpose, unitriangular)

Description du paramètre

  • A: matrice de coefficients (doit être une matrice carrée)
  • B: matrice ou vecteur du second membre
  • upper: si A est une matrice triangulaire supérieure (par défaut True)
  • transpose: s'il faut transposer A (par défaut False)
  • unitriangular: s'il faut utiliser une matrice triangulaire unitaire (par défaut False)

Exemple d'utilisation

Exemple

import torch

# Créer une matrice de coefficients triangulaire supérieure
A = torch.tensor([[3.0, 1.0, 2.0],
                  [0.0, 2.0, 1.0],
                  [0.0, 0.0, 1.0]])

# Vecteur du second membre
B = torch.tensor([9.0, 5.0, 2.0])

# Résoudre AX = B
X = torch.triangular_solve(B.unsqueeze(1), A, upper=True)

print("Solution X:")
print(X.solution)

La sortie est :

解 X:
tensor([[1.],
        [2.],
        [2.]])

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

Autres extensions