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)