jax.numpy.cumulative_prod

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

沿陣列軸的累積乘積。

numpy.cumulative_prod() 的 JAX 實作。

參數:

• x (ArrayLike) – N 維陣列

• axis (int | None | None) – 沿其累積的整數軸。如果 x 是一維的，則此引數為選用項目，預設為零。

• dtype (DTypeLike | None | None) – 輸出的選用 dtype。

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