cumsum¶
沿给定轴(axis)的元素的累加和。默认结果的第一个元素和输入的第一个元素一致。如果exlusive为True,结果的第一个元素则为0。
参数¶
x (Variable) - 累加的输入,需要进行累加操作的变量Tensor/LoDTensor。
axis (int,可选) - 指明需要累加的维。-1代表最后一维。默认为:-1。
exclusive (bool,可选) - 是否执行exclusive累加。默认为:False。
reverse (bool,可选) - 若为True,则以相反顺序执行累加。默认为:False。
返回¶
Variable(Tensor)。是累加的结果,即累加器的输出。
返回类型¶
变量(Variable)。
代码示例¶
import paddle.fluid as fluid
data = fluid.layers.data(name="input", shape=[32, 784])
result = fluid.layers.cumsum(data, axis=0)