digamma¶
逐元素计算输入 Tensor 的 digamma 函数值
\[\begin{split}\\Out = \Psi(x) = \frac{ \Gamma^{'}(x) }{ \Gamma(x) }\\\end{split}\]
返回¶
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]])