Fonction torch.polygamma de PyTorch
Manuel de référence PyTorch torch
torch.polygammaest une fonction dans PyTorch utilisée pour calculer la fonction polygamma élément par élément.
Définition de fonction
torch.polygamma(n, input, out=None)
Exemple d'utilisation
Exemple
import torch
# Calculer la fonction polygamma
x = torch.tensor([1.0, 2.0, 3.0])
n = 1
result = torch.polygamma(n, x)
print(result)
# Calculer la fonction polygamma
x = torch.tensor([1.0, 2.0, 3.0])
n = 1
result = torch.polygamma(n, x)
print(result)
Autres extensions