View source on GitHub |
Returns the cumulative sum of elements along a given axis.
tf.keras.ops.cumsum(
x, axis=None, dtype=None
)
Args | |
---|---|
x
|
Input tensor. |
axis
|
Axis along which the cumulative sum is computed. By default the input is flattened. |
dtype
|
dtype of returned tensor. Defaults to x.dtype. |
Returns | |
---|---|
Output tensor. |