You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.ops.TensorScatterSub.rst 1.5 kB

123456789101112131415161718192021222324252627
  1. mindspore.ops.TensorScatterSub
  2. ===============================
  3. .. py:class:: mindspore.ops.TensorScatterSub
  4. 根据指定的更新值和输入索引,通过减法运算更新输入Tensor的值。当同一个索引有多个不同值时,更新的结果将分别减去这些值。此操作几乎等同于使用 :class:`mindspore.ops.ScatterNdSub` ,只是更新后的结果是通过算子output返回,而不是直接原地更新input。
  5. `indices` 的最后一个轴是每个索引向量的深度。对于每个索引向量, `updates` 中必须有相应的值。`updates` 的shape应该等于 `input_x[indices]` 的shape。有关更多详细信息,请参见使用用例。
  6. .. note::
  7. 如果 `indices` 的某些值超出范围,则相应的 `updates` 不会更新为 `input_x` ,而不是抛出索引错误。
  8. **输入:**
  9. - **input_x** (Tensor) - 输入Tensor。 `input_x` 的维度必须不小于indices.shape[-1]。
  10. - **indices** (Tensor) - 输入Tensor的索引,数据类型为int32或int64的,其rank必须至少为2。
  11. - **updates** (Tensor) - 指定与 `input_x` 相减操作的Tensor,其数据类型与输入相同。updates.shape应等于indices.shape[:-1] + input_x.shape[indices.shape[-1]:]。
  12. **输出:**
  13. Tensor,shape和数据类型与输入 `input_x` 相同。
  14. **异常:**
  15. - **TypeError** - `indices` 的数据类型既不是int32,也不是int64。
  16. - **ValueError** - `input_x` 的shape长度小于 `indices` 的shape的最后一个维度。