PyTorch torch.digamma-Funktion

PyTorch torch.digamma-Funktion


Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.digammaEs ist eine Funktion in PyTorch, die für die elementweise Berechnung der Psi-Funktion (logarithmische Ableitung) verwendet wird.

Funktionsdefinition

torch.digamma(input, *, out=None)

Anwendungsbeispiel

Beispiel

import torch

# Beispiel für die Berechnung der Psi-Funktion
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])

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen