jax.numpy.poly

jax.numpy.poly(seq_of_zeros)

返回给定根序列的多项式系数。

JAX实现的 numpy.poly()。

参数:

seq_of_zeros (ArrayLike) – 一个标量或形状为 (M,) 或 (M, M) 的多项式根的数组。

返回:

一个包含多项式系数的数组。输出的数据类型总是提升为非精确类型。

返回类型:

Array

注意

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)