jax.numpy.polyder

内容

jax.numpy.polyder#

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

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

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)