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