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)。

代码示例

  1. import paddle.fluid as fluid
  2. data = fluid.layers.data(name="input", shape=[32, 784])
  3. result = fluid.layers.cumsum(data, axis=0)