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]