jax.numpy.polyder

jax.numpy.polyder(p, m=1)

返回指定阶数导数的系数。

JAX 对 numpy.polyder() 的实现。

参数:

• p (ArrayLike) – 多项式系数数组。

• m (int) – 求导阶数（正整数）。默认为 1。必须静态指定。

返回:

表示导数的多项式系数数组。

返回类型:

Array

注意

jax.numpy.polyder() 在输入为整数数组时与 numpy.polyder() 不同。NumPy 返回结果的 dtype 为 int，而 JAX 返回结果的 dtype 为 float。

另请参阅

• jax.numpy.polyint()：计算多项式的积分。

• jax.numpy.polyval()：在特定值处计算多项式。

示例

多项式 \(2 x^3 - 5 x^2 + 3 x - 1\) 的一阶导数是 \(6 x^2 - 10 x +3\)

>>> p = jnp.array([2, -5, 3, -1])
>>> jnp.polyder(p)
Array([  6., -10.,   3.], dtype=float32)

其二阶导数是 \(12 x - 10\)

>>> jnp.polyder(p, m=2)
Array([ 12., -10.], dtype=float32)