polygamma

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

对于给定的 x 逐元素计算每个元素的多伽马函数值,其中 x 的大小无特殊限制。返回一个多伽马函数上的 Tensor。

\[\Phi^n(x) = \frac{d^n}{dx^n} [\ln(\Gamma(x))]\]

参数

  • x (Tensor) – 输入是一个多维的 Tensor,它的数据类型可以是 float32,float64。

  • n (int) - 指定需要求解 n 阶多伽马函数,它的数据类型是 int。

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

返回

  • Tensor (Tensor):在 x 处的多伽马函数的值。

代码示例

import paddle

data = paddle.to_tensor([2, 3, 25.5], dtype='float32')
res = paddle.polygamma(data, 1)
print(res)
# Tensor(shape=[2], dtype=float32, place=CUDAPlace(0), stop_gradient=True,
#       [0.64493407,  0.39493407,  0.03999467])