mindspore.ops.Polygamma

View Source On Gitee
class mindspore.ops.Polygamma[source]

Computes the \(a\).

Warning

This is an experimental API that is subject to change or deletion.

Refer to mindspore.ops.polygamma() for more details.

Inputs:

  • a (Tensor) - The order of the polygamma function, it has shape \(()\), supported types: int32, int64.

  • x (Tensor) - The tensor to compute the \(a\)-th derivative of the polygamma function with, supported types: float16, float32, float64.

Outputs:

Tensor, has the same dtype as x.

Supported Platforms:

GPU CPU

Examples

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([1.0, -0.5]), mindspore.float32)
>>> a = Tensor(np.array(1), mindspore.int64)
>>> polygamma = ops.Polygamma()
>>> output = polygamma(a, x)
>>> print(output)
[1.644934 8.934802]
>>> a = Tensor(np.array(2), mindspore.int64)
>>> output = polygamma(a, x)
>>> print(output)
[-2.404114  -0.8287967]
>>> a = Tensor(np.array(3), mindspore.int64)
>>> output = polygamma(a, x)
>>> print(output)
[  6.4939404 193.40909  ]
>>> a = Tensor(np.array(4), mindspore.int64)
>>> output = polygamma(a, x)
>>> print(output)
[-24.886265   -3.4742498]