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]