jax.numpy.cumulative_prod#
- jax.numpy.cumulative_prod(x, /, *, axis=None, dtype=None, include_initial=False)[源代码]#
沿数组轴的累积乘积。
JAX 实现的
numpy.cumulative_prod()
。- 参数:
- 返回:
包含累积值的数组。
- 返回类型:
另请参阅
jax.numpy.cumprod()
:累积乘积的替代 API。jax.numpy.nancumprod()
:累积乘积,同时忽略 NaN 值。jax.numpy.multiply.accumulate()
:通过 ufunc API 的累积乘积。
示例
>>> x = jnp.array([[1, 2, 3], ... [4, 5, 6]]) >>> jnp.cumulative_prod(x, axis=1) Array([[ 1, 2, 6], [ 4, 20, 120]], dtype=int32) >>> jnp.cumulative_prod(x, axis=1, include_initial=True) Array([[ 1, 1, 2, 6], [ 1, 4, 20, 120]], dtype=int32)