Cumsum

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

• 算子功能：对输入张量self的元素，按照指定维度dim依次进行累加，并将结果保存到输出张量out中。
• 计算公式：$x\_i$是输入张量self中，从维度dim视角来看的某个元素（其它维度下标不变，只dim维度下标依次递增），$y\_i$是输出张量out中对应位置的元素，则：

$$y_i=x_1+x_2+x_3+......+x_i$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入Tensor。	INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64 FLOAT16、FLOAT、DOUBLE、COMPLEX64、COMPLEX128	ND
axis	输入	需要进行依次累加的维度。	INT32、INT64	ND
exclusive	属性	默认值为false，表示执行包含性累积求和（inclusive cumsum），即输出的第一个元素与输入的第一个元素相同； true 表示执行排除性累计求和（exclusive cumsum），即输出的第一个元素为0，后续元素为输入的前缀和。	BOOL	-
reverse	属性	默认值为false，表示从张量的开头向末尾进行累积求和（正向计算）； true 表示从张量的末尾向开头进行累积求和（反向计算）。	BOOL	-
y	输出	按指定维度累加后的输出Tensor。	输入Tensor相同x	ND

约束说明

• 无。

调用说明

调用方式	样例代码	说明
aclnn调用	test_aclnn_cumsum_v2	通过aclnnCumsumV2接口方式调用cumsum算子。
图模式调用	test_geir_cumsum	通过算子IR构图方式调用cumsum算子。