jax.numpy.polyder#

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

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

JAX实现的 numpy.polyder() 函数。

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

  • 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)