Функция torch.pinverse в PyTorch


Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.pinverseявляется функцией в PyTorch, используемой для вычисления псевдообратной матрицы Мура — Пенроуза. Онаtorch.linalg.pinvявляется псевдонимом для.

Определение функции

torch.pinverse(input, rcond=1e-15)

Parameters:

  • input(Tensor): Входная матрица.
  • rcond(float, необязательно): Пороговое значение усечения сингулярных значений. По умолчанию: 1e-15.

Возвращаемое значение:

  • 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.pinverse(A)

print("Матрица A:")
print(A)
print("nПсевдообратная A^+:")
print(A_pinv)

Результат вывода:

矩阵 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)

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения