mindspore.ops.polygamma
- mindspore.ops.polygamma(n, input)[源代码]
计算关于 input 的多伽马函数的 \(n\) 阶导数。
\[\psi^{(a)}(x) = \frac{d^{(a)}}{dx^{(a)}} \psi(x)\]其中 \(\psi(x)\) 为digamma函数。
- 参数:
n (Tensor) - 多伽马函数求导的阶数,支持的数据类型为int32和int64, n 的shape为 \(()\) 。
input (Tensor) - 用于计算多伽马函数的 \(n\) 阶导数。
- 返回:
Tensor。数据类型与 input 一致。
- 异常:
TypeError - input 不是Tensor。
TypeError - input 的数据类型不是float16、float32或float64。
TypeError - n 的数据类型不是int32或int64。
TypeError - n 的shape不是 \(()\) 。
- 支持平台:
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> x = Tensor(np.array([3.14, -2.71]), mindspore.float64) >>> a = Tensor(np.array(1), mindspore.int64) >>> output = ops.polygamma(a, x) >>> print(output) [ 0.37446456 15.49884838]