20 lines
1.1 KiB
ReStructuredText
20 lines
1.1 KiB
ReStructuredText
mindspore.ops.ScatterNdSub
|
||
===========================
|
||
|
||
.. py:class:: mindspore.ops.ScatterNdSub(use_locking=False)
|
||
|
||
使用给定值通过减法运算和输入索引更新Tensor值。在更新完成后输出 `input_x` ，这有利于更加方便地使用更新后的值。
|
||
|
||
更多参考详见 :func:`mindspore.ops.scatter_nd_sub`。
|
||
|
||
参数：
|
||
- **use_locking** (bool，可选) - 是否启用锁保护。默认值： ``False`` 。
|
||
|
||
输入：
|
||
- **input_x** (Union[Parameter, Tensor]) - 输入参数，数据类型为Parameter或Tensor，其shape为 :math:`(N, *)` ，其中 :math:`*` 为任意数量的额外维度。
|
||
- **indices** (Tensor) - 指定减法操作的索引，数据类型为mindspore.int32。索引的rank必须至少为2，并且 `indices.shape[-1] <= len(shape)` 。
|
||
- **updates** (Tensor) - 指定与 `input_x` 进行减法操作的Tensor，数据类型与 `input_x` 相同，shape为 `indices.shape[:-1] + input_x.shape[indices.shape[-1]:]` 。
|
||
|
||
输出：
|
||
Tensor，更新后的 `input_x` ，shape和数据类型与 `input_x` 相同。
|