PyTorch función torch.linalg.inv

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.linalg.invEs la función del módulo de álgebra lineal de PyTorch utilizada para calcular la inversa de una matriz. Estorch.inversela alternativa recomendada a.

Definición de la función

torch.linalg.inv(A)

Parámetros:

  • A(Tensor): Matriz cuadrada de entrada, debe ser invertible.

Valor de retorno:

  • torch.Tensor: Devuelve la inversa de la matriz.

Ejemplo de uso

Ejemplo

import torch

# Crear una matriz invertible
A = torch.tensor([[1.0, 2.0],
                  [3.0, 4.0]])

# Calcular la matriz inversa
A_inv = torch.linalg.inv(A)

print("Matriz A:")
print(A)
print("nMatriz inversa A^-1:")
print(A_inv)
print("nVerificación: A @ A^-1 =")
print(A @ A_inv)

El resultado de salida es:

矩阵 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 参考手册Manual de referencia de Pytorch torch

Otras extensiones