jax.numpy.poly

jax.numpy.poly(seq_of_zeros)

返回給定根序列的多項式係數。

JAX 實現的 numpy.poly()。

參數：

seq_of_zeros (ArrayLike) – 多項式根的純量或陣列，形狀為 (M,) 或 (M, M)。

返回：

一個包含多項式係數的陣列。輸出的 dtype 始終提升為非精確型別。

返回型別：

陣列

注意

jax.numpy.poly() 與 numpy.poly() 不同

• 當輸入為純量時，np.poly 會引發 TypeError，而 jnp.poly 將純量視為與長度為 1 的陣列相同。

• 對於複數值或方形輸入，jnp.poly 始終返回複數係數，而 np.poly 可能會根據其值返回實數或複數。

另請參閱

• jax.numpy.polyfit()：最小平方多項式擬合。

• jax.numpy.polyval()：在特定值評估多項式。

• jax.numpy.roots()：計算給定係數多項式的根。

範例

純量輸入

>>> jnp.poly(1)
Array([ 1., -1.], dtype=float32)

具有整數值的輸入陣列

>>> x = jnp.array([1, 2, 3])
>>> jnp.poly(x)
Array([ 1., -6., 11., -6.], dtype=float32)

具有複共軛的輸入陣列

>>> x = jnp.array([2, 1+2j, 1-2j])
>>> jnp.poly(x)
Array([  1.+0.j,  -4.+0.j,   9.+0.j, -10.+0.j], dtype=complex64)

作為具有實數值輸入的方矩陣的輸入陣列

>>> x = jnp.array([[2, 1, 5],
...                [3, 4, 7],
...                [1, 3, 5]])
>>> jnp.round(jnp.poly(x))
Array([  1.+0.j, -11.-0.j,   9.+0.j, -15.+0.j], dtype=complex64)