tf.keras.ops.nancumprod

Returns the cumulative product of elements along a given axis, treating NaNs as one.

View aliases

Main aliases

tf.keras.ops.numpy.nancumprod

tf.keras.ops.nancumprod(
    x, axis=None, dtype=None
)

Args
x	Input tensor.
axis	Axis along which the cumulative product is computed. By default the input is flattened.
dtype	dtype of returned tensor. Defaults to x.dtype.

Returns
Output tensor.

Examples:

import numpy as np
from keras import ops
x = np.array([[1.0, np.nan, 3.0],
              [np.nan, 2.0, 1.0]])
ops.nancumprod(x)
array([1., 1., 3., 3., 6., 6.])

ops.nancumprod(x, axis=1)
array([[1., 1., 3.],
       [1., 2., 2.]])