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

         