Función torch.polygamma de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.polygammaEs una función de PyTorch utilizada para calcular la función polygamma elemento por elemento.

Definición de la función

torch.polygamma(n, input, out=None)

Ejemplo de uso

Ejemplo

import torch

# Calcula la función polygamma
x = torch.tensor([1.0, 2.0, 3.0])
n = 1
result = torch.polygamma(n, x)
print(result)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones