19 lines
1.1 KiB
ReStructuredText
19 lines
1.1 KiB
ReStructuredText
mindspore.ops.TensorScatterMul
|
||
==============================
|
||
|
||
.. py:class:: mindspore.ops.TensorScatterMul
|
||
|
||
根据指定的更新值 `updates` 和输入索引 `indices` ，使用乘法运算更新输入Tensor的值。当同一索引有不同更新值时，更新的结果将是累积乘法的结果。此操作与 :class:`mindspore.ops.ScatterNdMul` 类似，但更新后的结果是返回一个新的输出Tensor，而不是直接更新 `input_x` 。
|
||
|
||
.. math::
|
||
output\left [indices \right ] = input\_x\times updates
|
||
|
||
更多参考详见： :func:`mindspore.ops.tensor_scatter_mul`。
|
||
|
||
输入：
|
||
- **input_x** (Tensor) - 输入Tensor。 `input_x` 的维度必须大于等于indices.shape[-1]。
|
||
- **indices** (Tensor) - `input_x` 执行scatter操作的目标索引，数据类型为int32或int64，rank必须大于等于2。
|
||
- **updates** (Tensor) - 指定与 `input_x` 相乘操作的Tensor，其数据类型与 `input_x` 相同。并且shape应等于 :math:`indices.shape[:-1] + input\_x.shape[indices.shape[-1]:]`。
|
||
|
||
输出：
|
||
Tensor，shape和数据类型与输入 `input_x` 相同。 |