TensorFlow函数 tf.cumsum及应用举例

     tf.cumsum

cumsum(
    x,
    axis=0,
    exclusive=False,
    reverse=False,
    name=None
)

函数 cumsum 是 cumulative sum缩写，计算累积和，即沿着tensor（张量）x的某一个维度axis，计算累积和。

参数解释：

— x, 即我们要计算累积和的tensor。

— axis=0, 默认是沿着x的第0维计算累积和。

— exclusive=False, 表示输出结果的第一元素是否与输入的第一个元素一致。默认exclusive=False，表示输出的第一个元素与输入的第一个元素一致（By default, this op performs an inclusive cumsum, which means that the first element of the input is identical to the first element of the output）。这是官方文档的解释。当我们对一个数组arr（或其他什么东东）进行累积求和时，我们要对累积和sum进行初始化，初始化的方式有两种，一种是将累积和初始化为0，即sum=0，一种是使用数组arr的第一个元素对累积和进行初始化，即sum=arr[0]。所以参数exclusive描述的是如何对累积和进行初始化。