mindspore.ops.Digamma

class mindspore.ops.Digamma

计算输入的lgamma函数的导数。

\[P(x) = grad(ln(gamma(x)))\]

警告

这是一个实验性API，后续可能修改或删除。

输入：

• x (Tensor) - 输入Tensor。数据类型为 float16 、 float32 或者 float64 。

输出：

Tensor, 和输入 x 具有相同的数据类型。

异常：

• TypeError - 如果输入 x 不是Tensor。

• TypeError - 输入输入 x 的数据类型不是 float16 、 float32 或者 float64 。

支持平台：

GPU CPU

样例：

>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([1.5, 0.5, 9]).astype(np.float16))
>>> digamma = ops.Digamma()
>>> output = digamma(x)
>>> print(output)
[ 0.0365 -1.964   2.14  ]