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]