jax.numpy.nancumprod

jax.numpy.nancumprod(a, axis=None, dtype=None, out=None)

沿著軸的元素累積乘積，忽略 NaN 值。

numpy.nancumprod() 的 JAX 實作。

參數:

• a (ArrayLike) – 要累積的 N 維陣列。

• axis (int | None) – 沿著其累積的整數軸。如果為 None (預設值)，則陣列將被展平並沿著展平的軸累積。

• dtype (DTypeLike | None) – 選擇性指定輸出的 dtype。如果未指定，則輸出 dtype 將與輸入 dtype 相符。

• out (None) – JAX 未使用

傳回值:

一個陣列，包含沿著給定軸的累積乘積。

傳回類型:

Array

另請參閱

• jax.numpy.cumprod()：不忽略 NaN 值的累積乘積。

• jax.numpy.multiply.accumulate()：透過 ufunc 方法的累積乘積。

• jax.numpy.prod()：沿軸的乘積

範例

>>> x = jnp.array([[1., 2., jnp.nan],
...                [4., jnp.nan, 6.]])

標準累積乘積將傳播 NaN 值

>>> jnp.cumprod(x)
Array([ 1.,  2., nan, nan, nan, nan], dtype=float32)

nancumprod() 將忽略 NaN 值，有效地將其替換為 1

>>> jnp.nancumprod(x)
Array([ 1.,  2.,  2.,  8.,  8., 48.], dtype=float32)

沿軸 1 的累積乘積

>>> jnp.nancumprod(x, axis=1)
Array([[ 1.,  2.,  2.],
       [ 4.,  4., 24.]], dtype=float32)