jax.numpy.roots

jax.numpy.roots(p, *, strip_zeros=True)

给定系数 p，返回多项式的根。

JAX 实现 numpy.roots()。

参数:

• p (ArrayLike) – 具有秩-1 的多项式系数数组。

• strip_zeros (bool) – 布尔值，默认为 True。如果为 True，则将删除系数中的前导零，类似于 numpy.roots()。如果设置为 False，则不会删除前导零，未定义的根将用函数输出中的 NaN 值表示。strip_zeros 必须设置为 False 才能使函数与 jax.jit() 和其他 JAX 转换兼容。

返回:

包含多项式根的数组。

返回类型:

Array

注意

与 np.roots 不同，此函数 jnp.roots 返回的根是复数数组，与根的值无关。

另请参阅

• jax.numpy.poly(): 查找给定根序列的多项式系数。

• jax.numpy.polyfit(): 数据多项式最小二乘拟合。

• jax.numpy.polyval(): 在特定值处求多项式。

示例

>>> coeffs = jnp.array([0, 1, 2])

默认行为与 numpy 匹配，并删除前导零

>>> jnp.roots(coeffs)
Array([-2.+0.j], dtype=complex64)

当 strip_zeros=False 时，额外的根将设置为 NaN

>>> jnp.roots(coeffs, strip_zeros=False)
Array([-2. +0.j, nan+nanj], dtype=complex64)