mindspore/docs/api/api_python/ops/mindspore.ops.CumSum.rst

24 lines
958 B
ReStructuredText
Raw Normal View History

mindspore.ops.CumSum
=====================
.. py:class:: mindspore.ops.CumSum(exclusive=False, reverse=False)
计算输入Tensor在指定轴上的累加和。
.. math::
y_i = x_1 + x_2 + x_3 + ...+ x_i
2022-07-22 14:07:38 +08:00
参数:
- **exclusive** (bool) - 表示输出结果的第一个元素是否与输入的第一个元素一致。如果为False表示输出的第一个元素与输入的第一个元素一致。默认值False。
- **reverse** (bool) - 如果为True则逆向计算累加和。默认值False。
2022-07-22 14:07:38 +08:00
输入:
- **input** (Tensor) - 输入要计算的Tensor。
- **axis** (int) - 指定要累加和的轴。仅支持常量值。该值在[-rank(input), rank(input))范围中。
2022-07-22 14:07:38 +08:00
输出:
Tensor。输出Tensor的shape与输入Tensor的shape一致。
2022-07-22 14:07:38 +08:00
异常:
- **TypeError** - `exclusive``reverse` 不是bool。
- **TypeError** - `axis` 不是int。