Fonction torch.linalg.inv de PyTorch

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

torch.linalg.invest une fonction du module d'algèbre linéaire de PyTorch utilisée pour calculer l'inverse d'une matrice. Elle esttorch.inversela fonction alternative recommandée.

Définition de fonction

torch.linalg.inv(A)

Parameters:

  • A(Tensor) : matrice carrée d'entrée, doit être inversible.

Valeur de retour:

  • torch.Tensor: Renvoie l'inverse de la matrice.

Exemple d'utilisation

Exemple

import torch

# Créer une matrice inversible
A = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]])

# Calculer la matrice inverse
A_inv = torch.linalg.inv(A)

print("Matrice A :")
print(A)
print("\nMatrice inverse A^-1 :")
print(A_inv)
print("\nVérification : A @ A^-1 =")
print(A @ A_inv)

Le résultat de sortie est:

矩阵 A:
tensor([[1., 2.],
        [3., 4.]])
逆矩阵 A^-1:
tensor([[-2.0000,  1.0000],
        [ 1.5000, -0.5000]])
验证: A @ A^-1 =
tensor([[1., 0.],
        [0., 1.]])

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

Autres extensions