Функция torch.polygamma в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.polygamma— это функция в PyTorch, используемая для поэлементного вычисления функции polygamma.

Определение функции

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

Пример использования

Пример

import torch

# Вычислить функцию polygamma
x = torch.tensor([1.0, 2.0, 3.0])
n = 1
result = torch.polygamma(n, x)
print(result)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения