cumsum
paddle.fluid.layers.
cumsum
(x, axis=None, exclusive=None, reverse=None)[源代码]
沿给定轴(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)