forked from mindspore-Ecosystem/mindspore
25 lines
1.1 KiB
ReStructuredText
25 lines
1.1 KiB
ReStructuredText
mindspore.ops.sum
|
||
==================
|
||
|
||
.. py:function:: mindspore.ops.sum(x, dim=None, keepdim=False, *, dtype=None)
|
||
|
||
计算Tensor指定维度元素的和。
|
||
|
||
参数：
|
||
- **x** (Tensor) - 输入Tensor。
|
||
- **dim** (Union[None, int, tuple(int), list(int)]) - 求和的维度。如果 `dim` 为None，对Tensor中的所有元素求和。
|
||
如果 `dim` 为int组成的tuple或list，将对tuple中的所有维度求和，取值范围必须在 :math:`[-x.ndim, x.ndim)` 。默认值：None。
|
||
- **keepdim** (bool) - 是否保持输出Tensor的维度，如果为True，保持对应的维度且长度为1。如果为False，不保持维度。默认值：False。
|
||
|
||
关键字参数：
|
||
- **dtype** (:class:`mindspore.dtype`, 可选) - 期望输出Tensor的类型。默认值：None。
|
||
|
||
返回：
|
||
Tensor， `x` 指定维度的和。
|
||
|
||
异常：
|
||
- **TypeError** - `x` 不是Tensor类型。
|
||
- **TypeError** - `dim` 类型不是int，tulpe(int)，list(int)或None。
|
||
- **ValueError** - `dim` 取值不在 :math:`[-x.ndim, x.ndim)` 范围。
|
||
- **TypeError** - `keepdim` 不是Tensor类型。
|