forked from huawei/mindspore2022
26 lines
1.5 KiB
ReStructuredText
26 lines
1.5 KiB
ReStructuredText
mindspore.ops.TensorScatterAdd
|
||
===============================
|
||
|
||
.. py:class:: mindspore.ops.TensorScatterAdd
|
||
|
||
根据指定的更新值和输入索引，通过相加运算更新输入Tensor的值。当同一索引有不同值时，更新的结果将是所有值的总和。此操作与 :class:`mindspore.ops.ScatterNdAdd` 类似，只是更新后的结果是通过算子output返回，而不是直接原地更新input。
|
||
|
||
`indices` 的最后一个轴是每个索引向量的深度。对于每个索引向量， `updates` 中必须有相应的值。 `updates` 的shape应该等于 `input_x[indices]` 的shape。有关更多详细信息，请参见使用用例。
|
||
|
||
.. note::
|
||
如果 `indices` 的某些值超出范围，则相应的 `updates` 不会更新为 `input_x` ，而不是抛出索引错误。
|
||
|
||
**输入：**
|
||
|
||
- **input_x** (Tensor) - 输入Tensor。 `input_x` 的维度必须不小于indices.shape[-1]。
|
||
- **indices** (Tensor) - 输入Tensor的索引，数据类型为int32或int64的。其rank必须至少为2。
|
||
- **updates** (Tensor) - 指定与 `input_x` 相加操作的Tensor，其数据类型与输入相同。updates.shape应等于indices.shape[:-1] + input_x.shape[indices.shape[-1]:]。
|
||
|
||
**输出：**
|
||
|
||
Tensor，shape和数据类型与输入 `input_x` 相同。
|
||
|
||
**异常：**
|
||
|
||
- **TypeError** - `indices` 的数据类型既不是int32，也不是int64。
|
||
- **ValueError** - `input_x` 的shape长度小于 `indices` 的shape的最后一个维度。 |