mindspore.ops.Digamma

class mindspore.ops.Digamma[源代码]

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

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

警告

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

输入:
  • x (Tensor) - 输入Tensor。数据类型为 float16float32 或者 float64

输出:

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

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

  • TypeError - 输入输入 x 的数据类型不是 float16float32 或者 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  ]