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.

Note

This forms part of the old polynomial API. Since version 1.4, the new polynomial API defined in numpy.polynomial is preferred. A summary of the differences can be found in the transition guide.

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