forked from mindspore-Ecosystem/mindspore
27 lines
1.3 KiB
ReStructuredText
27 lines
1.3 KiB
ReStructuredText
|
mindspore.ops.TensorScatterMin
|
|||
|
|
===============================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.ops.TensorScatterMin
|
|||
|
|
|
|||
|
|
根据指定的更新值和输入索引，通过最小值运算更新输入Tensor的值。
|
|||
|
|
|
|||
|
|
索引的最后一个轴是每个索引向量的深度。对于每个索引向量， `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的最后一个维度。
|