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)]) - 求和的维度。如果 `dim` 为None，对Tensor中的所有元素求和。
|
|||
|
|
如果 `dim` 为int组成的tuple，将对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` 不是Tensor类型。
|
|||
|
|
- **ValueError** - `dim` 取值不在 :math:`[-x.ndim, x.ndim)` 范围。
|
|||
|
|
- **TypeError** - `keepdim` 不是Tensor类型。
|