mindspore.ops.digamma

mindspore.ops.digamma(input)[源代码]

计算对数gamma函数在输入上的导数。

\[P(x) = \frac{d}{dx}(\ln (\Gamma(x)))\]

警告

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

参数:
  • input (Tensor) - 输入Tensor,数据类型是float16,float32或float64。

返回:

Tensor,数据类型和 input 一样。

异常:
  • TypeError - input 不是Tensor。

  • TypeError - input 的数据类型不是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))
>>> output = ops.digamma(x)
>>> print(output)
[ 0.0365 -1.964   2.14  ]