Fonction torch.pinverse de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.pinverseest une fonction de PyTorch utilisée pour calculer la pseudo-inverse de Moore-Penrose d'une matrice. Elle esttorch.linalg.pinvun alias de.

Définition de fonction

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

Parameters:

  • input(Tensor): matrice d'entrée.
  • rcond(float, facultatif) : Seuil de troncature des valeurs singulières. Valeur par défaut : 1e-15.

Valeur de retour:

  • torch.Tensor: Renvoie le pseudo-inverse de la matrice.

Exemple d'utilisation

Exemple

import torch

# Créer une matrice non carrée
A = torch.tensor([[1.0, 2.0, 3.0],
                  [4.0, 5.0, 6.0]], dtype=torch.float64)

# Calculer le pseudo-inverse
A_pinv = torch.pinverse(A)

print("Matrice A :")
print(A)
print("nPseudo-inverse A^+:")
print(A_pinv)

Le résultat de sortie est:

矩阵 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 参考手册Manuel de référence Pytorch torch

Autres extensions