Python PyTorch digamma用法及代碼示例

本文簡要介紹python語言中 torch.special.digamma 的用法。

用法:

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

參數：

input(Tensor) -計算 digamma 函數的張量

關鍵字參數：

out(Tensor,可選的) -輸出張量。

計算 input 上 gamma 函數的對數導數。

\digamma(x) = \frac{d}{dx} \ln\left(\Gamma\left(x\right)\right) = \frac{\Gamma'(x)}{\Gamma(x)}

注意

這個函數類似於 SciPy 的 scipy.special.digamma 。

注意

從 PyTorch 1.8 開始，digamma 函數為 0 返回 -Inf 。以前它為 0 返回 NaN 。

例子：

>>> a = torch.tensor([1, 0.5])
>>> torch.special.digamma(a)
tensor([-0.5772, -1.9635])