jax.numpy.polymul

jax.numpy.polymul(a1, a2, *, trim_leading_zeros=False)

返回两个多项式的乘积。

JAX 中 numpy.polymul() 的实现。

参数:

• a1 (ArrayLike) – 多项式系数的一维数组。

• a2 (ArrayLike) – 多项式系数的一维数组。

• trim_leading_zeros (bool) – 默认值为 False。如果为 True，则会删除返回值中的前导零，以匹配 numpy 的结果。但这会阻止该函数在编译后的代码中使用。由于浮点算术误差累积的差异，被视为零的值的截止值可能会导致 NumPy 和 JAX 之间，甚至不同 JAX 后端之间的结果不一致。当 trim_leading_zeros=True 时，结果可能会导致输出形状不一致。

返回:

两个多项式乘积的系数数组。输出的 dtype 始终被提升为非精确类型。

返回类型:

数组

注意

与 numpy.polymul() 接受标量输入不同，jax.numpy.polymul() 仅接受数组作为输入。

另请参阅

• jax.numpy.polyadd()：计算两个多项式的和。

• jax.numpy.polysub()：计算两个多项式的差。

• jax.numpy.polydiv()：计算多项式除法的商和余数。

示例

>>> x1 = np.array([2, 1, 0])
>>> x2 = np.array([0, 5, 0, 3])
>>> np.polymul(x1, x2)
array([10,  5,  6,  3,  0])
>>> jnp.polymul(x1, x2)
Array([ 0., 10.,  5.,  6.,  3.,  0.], dtype=float32)

如果 trim_leading_zeros=True，则结果与 np.polymul 的结果匹配。

>>> jnp.polymul(x1, x2, trim_leading_zeros=True)
Array([10.,  5.,  6.,  3.,  0.], dtype=float32)

对于 dtype 为 complex 的输入数组

>>> x3 = np.array([2., 1+2j, 1-2j])
>>> x4 = np.array([0, 5, 0, 3])
>>> np.polymul(x3, x4)
array([10. +0.j,  5.+10.j, 11.-10.j,  3. +6.j,  3. -6.j])
>>> jnp.polymul(x3, x4)
Array([ 0. +0.j, 10. +0.j,  5.+10.j, 11.-10.j,  3. +6.j,  3. -6.j],      dtype=complex64)
>>> jnp.polymul(x3, x4, trim_leading_zeros=True)
Array([10. +0.j,  5.+10.j, 11.-10.j,  3. +6.j,  3. -6.j], dtype=complex64)