jax.numpy.poly

内容

jax.numpy.poly#

jax.numpy.poly(seq_of_zeros)[source]#

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

JAX 实现 numpy.poly().

参数：

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

返回值：

包含多项式系数的数组。输出的数据类型始终提升为不精确类型。

返回类型：

数组

注意

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

  • 当输入为标量时，np.poly 抛出 TypeError，而 jnp.poly 将标量视为长度为 1 的数组。

  • 对于复数或方形输入，jnp.poly 始终返回复数系数，而 np.poly 根据其值可能返回实数或复数。

另请参阅

示例

标量输入

>>> 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)
内容