PaddlePaddle

 paddle / cumsum


cumsum

paddle. cumsum ( x, axis=None, dtype=None, name=None ) [源代码]

沿给定 axis 计算张量 x 的累加和。

注意:结果的第一个元素和输入的第一个元素相同。

参数

  • x (Tensor) - 累加的输入,需要进行累加操作的Tensor.

  • axis (int,可选) - 指明需要累加的维度。-1代表最后一维。默认:None,将输入展开为一维变量再进行累加计算。

  • dtype (str,可选) - 输出Tensor的数据类型,支持int32、int64、float32、float64. 如果指定了,那么在执行操作之前,输入张量将被转换为dtype. 这对于防止数据类型溢出非常有用。默认为:None.

  • name (str,可选)- 操作的名称(可选,默认值为None)。更多信息请参见 Name

返回

Tensor, 累加的结果。

代码示例

import paddle

data = paddle.arange(12)
data = paddle.reshape(data, (3, 4))

y = paddle.cumsum(data)
# [ 0  1  3  6 10 15 21 28 36 45 55 66]

y = paddle.cumsum(data, axis=0)
# [[ 0  1  2  3]
#  [ 4  6  8 10]
#  [12 15 18 21]]

y = paddle.cumsum(data, axis=-1)
# [[ 0  1  3  6]
#  [ 4  9 15 22]
#  [ 8 17 27 38]]

y = paddle.cumsum(data, dtype='float64')
print(y.dtype)
# VarType.FP64

此页内容是否对您有帮助