maxframe.tensor.special.polygamma#

maxframe.tensor.special.polygamma(a, b, **kwargs)[源代码]#

多伽马函数。

定义为 \(\psi^{(n)}(x)\)，其中 \(\psi\) 是 digamma 函数。详见 [dlmf]。

参数:

  • n (array_like) -- digamma 函数的导数阶数；必须为整数

  • x (array_like) -- 实数输入

返回:

函数结果

返回类型:

ndarray

参见

digamma()

备注

Array API Standard Support

polygamma has experimental support for Python Array API Standard compatible backends in addition to NumPy. Please consider testing these features by setting an environment variable SCIPY_ARRAY_API=1 and providing CuPy, PyTorch, JAX, or Dask arrays as array arguments. The following combinations of backend and device (or other capability) are supported.

Library

CPU

GPU

NumPy

✅

n/a

CuPy

n/a

✅

PyTorch

✅

✅

JAX

✅

✅

Dask

✅

n/a

See Support for the array API standard for more information.

引用

[dlmf]

NIST, 数学函数数字图书馆, https://dlmf.nist.gov/5.15