mindspore.ops.digamma

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

计算lgamma对数函数在输入上的梯度。

\[P(input) = grad(In(gamma(input)))\]

警告

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

参数：

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

返回：

Tensor，数据类型和 input 一样。

异常：

  • TypeError - input 不是Tensor。

  • TypeError - input 的数据类型不是float16，float32或float64。

支持平台：

GPU CPU

样例：

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