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])