Python PyTorch polygamma用法及代码示例

本文简要介绍python语言中 torch.special.polygamma 的用法。

用法:

torch.special.polygamma(n, input, *, out=None) → Tensor

参数：

• n(int) -多伽马函数的阶

• input(Tensor) -输入张量。

关键字参数：

out(Tensor,可选的) -输出张量。

计算 input 上 digamma 函数的 n^{th} 导数。 n \geq 0 称为多伽玛函数的阶数。

\psi^{(n)}(x) = \frac{d^{(n)}}{dx^{(n)}} \psi(x)

注意

此函数仅对非负整数 n \geq 0 实现。

例子：

>>> a = torch.tensor([1, 0.5])
>>> torch.special.polygamma(1, a)
tensor([1.64493, 4.9348])
>>> torch.special.polygamma(2, a)
tensor([ -2.4041, -16.8288])
>>> torch.special.polygamma(3, a)
tensor([ 6.4939, 97.4091])
>>> torch.special.polygamma(4, a)
tensor([ -24.8863, -771.4742])