PyTorch torch.digamma función
Manual de referencia de Pytorch torch
torch.digammaEs una función en PyTorch que se usa para calcular elemento por elemento la función psi (derivada logarítmica).
Definición de la función
torch.digamma(input, *, out=None)
Ejemplo de uso
Ejemplo
import torch
# Ejemplo de cálculo de la función 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])
# Ejemplo de cálculo de la función 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])
Otras extensiones