mindspore.ops.polygamma

mindspore.ops.polygamma(n, input)

Computes the $n$-th derivative of the polygamma function on input.

$${\psi }^{(a)}(x)=\frac{d^{(a)}}{d{x}^{(a)}}\psi (x)$$

where $\psi (x)$ is the digamma function.

Parameters

• n (Tensor) – The order of the polygamma function.

• input (Tensor) – The input tensor.

Returns

Tensor

Supported Platforms:

GPU CPU

Examples

>>> import mindspore
>>> x = mindspore.tensor([3.14, -2.71], mindspore.float64)
>>> a = mindspore.tensor(1, mindspore.int64)
>>> output = mindspore.ops.polygamma(a, x)
>>> print(output)
[ 0.37446456 15.49884838]