PyTorch torch.polygamma-Funktion
PyTorch torch-Referenzhandbuch
torch.polygammaist eine Funktion in PyTorch, die zur elementweisen Berechnung der Polygamma-Funktion verwendet wird.
Funktionsdefinition
torch.polygamma(n, input, out=None)
Anwendungsbeispiel
Beispiel
import torch
# Polygamma-Funktion berechnen
x = torch.tensor([1.0, 2.0, 3.0])
n = 1
result = torch.polygamma(n, x)
print(result)
# Polygamma-Funktion berechnen
x = torch.tensor([1.0, 2.0, 3.0])
n = 1
result = torch.polygamma(n, x)
print(result)
Andere Erweiterungen