Функция PyTorch torch.linalg.inv
Справочное руководство Pytorch torch
torch.linalg.invявляется функцией в модуле линейной алгебры PyTorch, используемой для вычисления обратной матрицы. Онаtorch.inverseрекомендуемая альтернативная функция.
Определение функции
torch.linalg.inv(A)
Parameters:
A(Tensor): Входная квадратная матрица, должна быть обратимой.
Возвращаемое значение:
torch.Tensor: Возвращает обратную матрицу.
Пример использования
Пример
import torch
# Создать обратимую матрицу
A = torch.tensor([[1.0, 2.0],
[3.0, 4.0]])
# Вычислить обратную матрицу
A_inv = torch.linalg.inv(A)
print("Матрица A:")
print(A)
print("\nОбратная матрица A^-1:")
print(A_inv)
print("\nПроверка: A @ A^-1 =")
print(A @ A_inv)
# Создать обратимую матрицу
A = torch.tensor([[1.0, 2.0],
[3.0, 4.0]])
# Вычислить обратную матрицу
A_inv = torch.linalg.inv(A)
print("Матрица A:")
print(A)
print("\nОбратная матрица A^-1:")
print(A_inv)
print("\nПроверка: A @ A^-1 =")
print(A @ A_inv)
Результат вывода:
矩阵 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.]])
Другие расширения