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
样例:
>>> 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 ]