后处理累积和计算。

图1 CumsumOperation

struct CumsumParam {
    SVector<int64_t> axes;
    bool exclusive = false;
    bool reverse =  false;
};

成员名称	描述
axes	指定axis轴(维度)上计算累加和，只能包含一个轴索引。 axes的值必须小于输入x的维度数。
exclusive	在某一个轴上的累加结果从第几个元素开始，默认为false。 true：从第一个元素开始。 false：从第0个元素开始。
reverse	正向累加或逆向累加，默认为false。 true：输出正向累加。 false：输出逆向累加。

成员名称

描述

axes

指定axis轴(维度)上计算累加和，只能包含一个轴索引。

axes的值必须小于输入x的维度数。

exclusive

在某一个轴上的累加结果从第几个元素开始，默认为false。

reverse

正向累加或逆向累加，默认为false。

参数	维度	数据类型	格式
x	[-1,…,-1] -1表示当前维度的大小没有约束。	float16	ND

参数

维度

数据类型

格式

x

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16

ND

参数	维度	数据类型	格式
output	[-1,…,-1] -1表示当前维度的大小没有约束。	float16	ND

参数

维度

数据类型

格式

output

[-1,…,-1]

-1表示当前维度的大小没有约束。

float16

ND