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)
print(res)
# Tensor(shape=[2, 2], dtype=float32, place=CUDAPlace(0), stop_gradient=True,
#       [[-0.57721591,  0.03648996],
#        [ nan       ,  5.32286835]])