jax.numpy.polyder#

jax.numpy.polyder(p, m=1)[source]#

返回指定阶数多项式导数的系数。

numpy.polyder() 的 JAX 实现。

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

  • m (int) – 微分阶数(正整数)。默认为 1。必须静态指定。

返回:

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

返回类型:

Array

注意

当给定整数数组时,jax.numpy.polyder()numpy.polyder() 不同。NumPy 返回 dtype 为 int 的结果,而 JAX 返回 dtype 为 float 的结果。

另请参阅

示例

多项式 \(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)