Fonction torch.linalg.det de PyTorch

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

torch.linalg.detIl s'agit d'une fonction utilisée pour calculer le déterminant d'une matrice dans le module d'algèbre linéaire de PyTorch. Elle esttorch.detla fonction alternative recommandée.

Définition de fonction

torch.linalg.det(A)

Parameters:

  • A(Tensor) : Matrice carrée d'entrée.

Valeur de retour:

  • torch.Tensor: Renvoie le déterminant de la matrice.

Exemple d'utilisation

Exemple

import torch

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

# Calculer le déterminant
det_A = torch.linalg.det(A)

print("Matrice A :")
print(A)
print("Déterminant :")
print(det_A)

Le résultat de sortie est:

矩阵 A:
tensor([[1., 2., 3.],
        [0., 4., 5.],
        [0., 0., 6.]])
行列式:
tensor(24.)

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

Autres extensions