25 lines
1.2 KiB
ReStructuredText
25 lines
1.2 KiB
ReStructuredText
mindspore.ops.ScatterNdMul
|
||
===========================
|
||
|
||
.. py:class:: mindspore.ops.ScatterNdMul(use_locking=False)
|
||
|
||
对张量中的单个值或切片应用稀疏乘法。
|
||
|
||
使用给定值通过乘法运算和输入索引更新Parameter或Tensor值。在更新完成后输出 `input_x` ，这有利于更加方便地使用更新后的值。
|
||
|
||
.. warning::
|
||
这是一个实验性API，后续可能修改或删除。
|
||
|
||
更多参考详见 :func:`mindspore.ops.scatter_nd_mul` 。
|
||
|
||
参数：
|
||
- **use_locking** (bool，可选) - 是否启用锁保护。默认值： ``False`` 。
|
||
|
||
输入：
|
||
- **input_x** (Union[Parameter, Tensor]) - 输入参数，数据类型为Parameter或Tensor。
|
||
- **indices** (Tensor) - 指定乘法操作的索引，数据类型为mindspore.int32或mindspore.int64。索引的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` 相同。
|