cupy.cumsum¶
-
cupy.
cumsum
(a, axis=None, dtype=None, out=None)[source]¶ Returns the cumulative sum of an array along a given axis.
- Parameters
a (cupy.ndarray) – Input array.
axis (int) – Axis along which the cumulative sum is taken. If it is not specified, the input is flattened.
dtype – Data type specifier.
out (cupy.ndarray) – Output array.
- Returns
The result array.
- Return type
See also