jax.numpy.polymul

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

Find the product of two polynomials.

LAX-backend implementation of polymul().

Setting trim_leading_zeros=True makes the output match that of numpy. But prevents the function from being able to be used in compiled code.

Original docstring below.

Finds the polynomial resulting from the multiplication of the two input polynomials. Each input must be either a poly1d object or a 1D sequence of polynomial coefficients, from highest to lowest degree.

Parameters

• a1 (array_like or poly1d object) – Input polynomials.

• a2 (array_like or poly1d object) – Input polynomials.

Returns

out – The polynomial resulting from the multiplication of the inputs. If either inputs is a poly1d object, then the output is also a poly1d object. Otherwise, it is a 1D array of polynomial coefficients from highest to lowest degree.

Return type

ndarray or poly1d object