jax.numpy.cumulative_prod

jax.numpy.cumulative_prod(x, /, *, axis=None, dtype=None, include_initial=False)

沿数组轴的累积乘积。

JAX 对 numpy.cumulative_prod() 的实现。

参数:

• x (ArrayLike) – N 维数组

• axis (int | None) – 用于累积的整数轴。如果 x 是一维的，此参数是可选的，默认为零。

• dtype (DTypeLike | None) – 可选的输出数据类型。

• include_initial (bool) – 如果为 True，则在累积乘积中包含初始值。默认为 False。

返回:

包含累积值的数组。

返回类型:

数组

另请参阅

• 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)