Функция torch.linalg.pinv в PyTorch
Справочное руководство Pytorch torch
torch.linalg.pinvЭто функция в модуле линейной алгебры PyTorch, используемая для вычисления псевдообратной матрицы Мура — Пенроуза. Для неквадратных или вырожденных матриц псевдообратная матрица является важным инструментом.
Определение функции
torch.linalg.pinv(A, rcond=None, hermitian=False, out=None)
Parameters:
A(Tensor): входная матрица.rcond(float, необязательно): Порог отсечения сингулярных значений.hermitian(bool, необязательно): Если True, предполагается, что матрица эрмитова. По умолчанию False.out(Tensor, необязательный): Выходной тензор.
Возвращаемое значение:
torch.Tensor: Возвращает псевдообратную матрицу.
Пример использования
Пример
import torch
# Создать неквадратную матрицу
A = torch.tensor([[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0]], dtype=torch.float64)
# Вычислить псевдообратную матрицу
A_pinv = torch.linalg.pinv(A)
print("Матрица A:")
print(A)
print("nПсевдообратная A^+:")
print(A_pinv)
print("nПроверка: A @ A^+ @ A ≈ A")
print(A @ A_pinv @ A)
# Создать неквадратную матрицу
A = torch.tensor([[1.0, 2.0, 3.0],
[4.0, 5.0, 6.0]], dtype=torch.float64)
# Вычислить псевдообратную матрицу
A_pinv = torch.linalg.pinv(A)
print("Матрица A:")
print(A)
print("nПсевдообратная A^+:")
print(A_pinv)
print("nПроверка: A @ A^+ @ A ≈ A")
print(A @ A_pinv @ A)
Результат вывода:
矩阵 A:
tensor([[1., 2., 3.],
[4., 5., 6.]], dtype=torch.float64)
伪逆 A^+:
tensor([[-0.9444, 0.4444],
[-0.2778, 0.2778],
[ 0.3889, 0.1111]], dtype=float64)
验证: A @ A^+ @ A ≈ A
tensor([[ 1.0000e+00, 2.0000e+00, 3.0000e+00],
[ 4.0000e+00, 5.0000e+00, 6.0000e+00]], dtype=float64)
Другие расширения