cupy.cumsum

cupy.cumsum(a, axis=None, dtype=None, out=None)

返回沿给定轴的数组的累积和。

参数:

• a (cupy.ndarray) – 输入数组。

• axis (int) – 计算累积和的轴。如果未指定，则输入会被展平。

• dtype – 数据类型说明符。

• out (cupy.ndarray) – 输出数组。

返回:

结果数组。

返回类型:

cupy.ndarray

另请参阅

numpy.cumsum()