digamma

paddle. digamma ( x, name=None ) [源代码]

逐元素计算输入 Tensor 的 digamma 函数值

\[\begin{split}\\Out = \Psi(x) = \frac{ \Gamma^{'}(x) }{ \Gamma(x) }\\\end{split}\]

参数

  • x (Tensor) – 输入 Tensor。数据类型为 float32,float64。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

Tensor, digamma 函数计算结果,数据类型和维度大小与输入一致。

代码示例

>>> import paddle

>>> data = paddle.to_tensor([[1, 1.5], [0, -2.2]], dtype='float32')
>>> res = paddle.digamma(data)
>>> res
Tensor(shape=[2, 2], dtype=float32, place=Place(cpu), stop_gradient=True,
[[-0.57721591,  0.03648996],
 [ nan       ,  5.32286835]])