jax.numpy.polyadd

jax.numpy.polyadd(a1, a2)

傳回兩個多項式的和。

JAX 版本的 numpy.polyadd() 實作。

參數：

• a1 (ArrayLike) – 多項式係數陣列。

• a2 (ArrayLike) – 多項式係數陣列。

傳回值：

包含輸入多項式之和的係數的陣列。

傳回類型：

陣列

注意

jax.numpy.polyadd() 與 numpy.polyadd() 不同，僅接受陣列作為輸入，後者也接受純量輸入。

另請參閱

• jax.numpy.polysub()：計算兩個多項式的差。

• jax.numpy.polymul()：計算兩個多項式的乘積。

• jax.numpy.polydiv()：計算多項式除法的商和餘數。

範例

>>> x1 = jnp.array([2, 3])
>>> x2 = jnp.array([5, 4, 1])
>>> jnp.polyadd(x1, x2)
Array([5, 6, 4], dtype=int32)

>>> x3 = jnp.array([[2, 3, 1]])
>>> x4 = jnp.array([[5, 7, 3],
...                 [8, 2, 6]])
>>> jnp.polyadd(x3, x4)
Array([[ 5,  7,  3],
       [10,  5,  7]], dtype=int32)

>>> x5 = jnp.array([1, 3, 5])
>>> x6 = jnp.array([[5, 7, 9],
...                 [8, 6, 4]])
>>> jnp.polyadd(x5, x6)  
Traceback (most recent call last):
...
ValueError: Cannot broadcast to shape with fewer dimensions: arr_shape=(2, 3) shape=(2,)
>>> x7 = jnp.array([2])
>>> jnp.polyadd(x6, x7)
Array([[ 5,  7,  9],
       [10,  8,  6]], dtype=int32)