Fonction torch.digamma de PyTorch
Manuel de référence Pytorch torch
torch.digammaIl s'agit d'une fonction de PyTorch utilisée pour le calcul élément par élément de la fonction psi (dérivée logarithmique).
Définition de fonction
torch.digamma(input, *, out=None)
Exemple d'utilisation
Exemple
import torch
# Exemple de calcul de la fonction psi
x = torch.tensor([1.0, 2.0, 3.0, 4.0], dtype=torch.float32)
result = torch.digamma(x)
print(result) # tensor([-0.5772, 0.4228, 1.2561, 2.0312])
# Exemple de calcul de la fonction psi
x = torch.tensor([1.0, 2.0, 3.0, 4.0], dtype=torch.float32)
result = torch.digamma(x)
print(result) # tensor([-0.5772, 0.4228, 1.2561, 2.0312])
Autres extensions