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