forked from mindspore-Ecosystem/mindspore
25 lines
834 B
ReStructuredText
25 lines
834 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)
|
||
|
||
其中 :math:`\psi(x)` 为digamma函数。
|
||
|
||
参数：
|
||
- **a** (Tensor) - 多伽马函数求导的阶数，支持的数据类型为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:`()` 。
|