mindspore.ops.Polygamma

class mindspore.ops.Polygamma[源代码]

计算关于 x 的多伽马函数的 \(a\) 阶导数。

警告

这是一个实验性API，后续可能修改或删除。

更多参考详见 mindspore.ops.polygamma()。

输入：

a (Tensor) - 多伽马函数求导的阶数， a 的shape为 \(()\) ，支持的数据类型为int32和int64。
x (Tensor) - 用于计算多伽马函数的 \(a\) 阶导数的Tensor，支持的数据类型为：float16、float32和float64。

输出：

Tensor。数据类型与 x 一致。

支持平台：

GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> x = Tensor(np.array([1.0, -0.5]), mindspore.float32)
>>> a = Tensor(np.array(1), mindspore.int64)
>>> polygamma = ops.Polygamma()
>>> output = polygamma(a, x)
>>> print(output)
[1.644934 8.934802]
>>> a = Tensor(np.array(2), mindspore.int64)
>>> output = polygamma(a, x)
>>> print(output)
[-2.404114  -0.8287967]
>>> a = Tensor(np.array(3), mindspore.int64)
>>> output = polygamma(a, x)
>>> print(output)
[  6.4939404 193.40909  ]
>>> a = Tensor(np.array(4), mindspore.int64)
>>> output = polygamma(a, x)
>>> print(output)
[-24.886265   -3.4742498]