PyTorch torch.linalg.inv Funktion


Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.linalg.invist eine Funktion im Modul für lineare Algebra von PyTorch, die zur Berechnung der Inversen einer Matrix verwendet wird. Sie isttorch.inversedie empfohlene alternative Funktion.

Funktionsdefinition

torch.linalg.inv(A)

Parameters:

  • A(Tensor): Quadratische Eingabematrix, muss invertierbar sein.

Rückgabewert:

  • torch.Tensor: Gibt die Inverse der Matrix zurück.

Anwendungsbeispiel

Beispiel

import torch

# Eine invertierbare Matrix erstellen
A = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]])

# Die inverse Matrix berechnen
A_inv = torch.linalg.inv(A)

print("Matrix A:")
print(A)
print("\nInverse Matrix A^-1:")
print(A_inv)
print("\nÜberprüfung: A @ A^-1 =")
print(A @ A_inv)

Das Ausgabeergebnis ist:

矩阵 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 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen