megengine.functional.cumsum

cumsum(inp, axis)

返回在给定轴axis上inp张量元素的累加和

参数

• inp (Tensor) – 输入张量。

• axis (int) – 要进行累加的维度

实际案例

from megengine import tensor
import megengine.functional as F

x = tensor([[1, 2, 3], [4, 5, 6]], "int32")
y = F.cumsum(x, 1)
print(y.numpy())

输出：

[[ 1  3  6]
[ 4  9 15]]

上一页

megengine.functional.std

下一页

megengine.functional.dot