mindspore.ops.polygamma

mindspore.ops.polygamma(n, input)

计算关于 input 的多伽马函数的 \(n\) 阶导数。

\[\psi^{(a)}(x) = \frac{d^{(a)}}{dx^{(a)}} \psi(x)\]

其中 \(\psi(x)\) 为digamma函数。

参数：

• n (Tensor) - 多伽马函数求导的阶数，支持的数据类型为int32和int64， n 的shape为 \(()\) 。

• input (Tensor) - 用于计算多伽马函数的 \(n\) 阶导数。

返回：

Tensor。数据类型与 input 一致。

异常：

• TypeError - input 不是Tensor。

• TypeError - input 的数据类型不是float16、float32或float64。

• TypeError - n 的数据类型不是int32或int64。

• TypeError - n 的shape不是 \(()\) 。

支持平台：

GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([3.14, -2.71]), mindspore.float64)
>>> a = Tensor(np.array(1), mindspore.int64)
>>> output = ops.polygamma(a, x)
>>> print(output)
[ 0.37446456 15.49884838]