forked from mindspore-Ecosystem/mindspore
25 lines
823 B
ReStructuredText
25 lines
823 B
ReStructuredText
|
mindspore.ops.polygamma
|
|||
|
|
=======================
|
|||
|
|
|
|||
|
|
.. py:function:: mindspore.ops.polygamma(a, x)
|
|||
|
|
|
|||
|
|
计算关于 `x` 的多伽马函数的 :math:`a` 阶导数。
|
|||
|
|
|
|||
|
|
.. math::
|
|||
|
|
\psi^{(a)}(x) = \frac{d^{(a)}}{dx^{(a)}} \psi(x)
|
|||
|
|
|
|||
|
|
其中\psi(x)为digamma函数。
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
- **a** (tuple) - 多伽马函数求导的阶数，支持的数据类型为int32和int64， `a` 的shape为 :math:`()` 。
|
|||
|
|
- **x** (Tensor) - 用于计算多伽马函数的Tensor。
|
|||
|
|
|
|||
|
|
返回：
|
|||
|
|
Tensor。数据类型与 `x` 一致。
|
|||
|
|
|
|||
|
|
异常：
|
|||
|
|
- **TypeError** - `x` 不是Tensor。
|
|||
|
|
- **TypeError** - `x` 的数据类型不是float16、float32或float64。
|
|||
|
|
- **TypeError** - `a` 的数据类型不是int32或int64。
|
|||
|
|
- **TypeError** - `a` 的shape不是 :math:`()` 。
|