jax.numpy.polyint

目录

jax.numpy.polyint#

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

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

JAX 实现的 numpy.polyint()。

参数:

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

  • m (int) – 积分顺序。默认值为1。必须静态指定。

  • k (int | ArrayLike | None) – 标量或 m 个积分常数的数组。

返回:

一个积分多项式的系数数组。

返回类型:

Array

参见

示例

多项式 \(12 x^2 + 12 x + 6\) 的一阶积分是 \(4 x^3 + 6 x^2 + 6 x\)。

>>> p = jnp.array([12, 12, 6])
>>> jnp.polyint(p)
Array([4., 6., 6., 0.], dtype=float32)

由于常数 k 未提供，结果末尾包含了 0。如果提供了常数 k：

>>> jnp.polyint(p, k=4)
Array([4., 6., 6., 4.], dtype=float32)

二阶积分是 \(x^4 + 2 x^3 + 3 x\):

>>> jnp.polyint(p, m=2)
Array([1., 2., 3., 0., 0.], dtype=float32)

当 m>=2 时，常数 k 应作为一个包含 m 个元素的数组提供。多项式 \(12 x^2 + 12 x + 6\) 在常数 k=[4, 5] 下的二阶积分是 \(x^4 + 2 x^3 + 3 x^2 + 4 x + 5\)：

>>> jnp.polyint(p, m=2, k=jnp.array([4, 5]))
Array([1., 2., 3., 4., 5.], dtype=float32)
目录