jax.numpy.polyder#

jax.numpy.polyder(p, m=1)[源代码]#

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

JAX 实现的 numpy.polyder()

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

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

返回:

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

返回类型:

数组

注意

jax.numpy.polyder() 在给定整数数组时与 numpy.polyder() 不同。NumPy 返回的结果数据类型为 int,而 JAX 返回的结果数据类型为 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)