jax.numpy.cumproductΒΆ

jax.numpy.cumproduct(a, axis=None, dtype=None, out=None)ΒΆ

Return the cumulative product of elements along a given axis.

LAX-backend implementation of cumprod().

Original docstring below.

Parameters
  • a (array_like) – Input array.

  • axis (int, optional) – Axis along which the cumulative product is computed. By default the input is flattened.

  • dtype (dtype, optional) – Type of the returned array, as well as of the accumulator in which the elements are multiplied. If dtype is not specified, it defaults to the dtype of a, unless a has an integer dtype with a precision less than that of the default platform integer. In that case, the default platform integer is used instead.

Returns

cumprod – A new array holding the result is returned unless out is specified, in which case a reference to out is returned.

Return type

ndarray