22 lines
1.1 KiB
ReStructuredText
22 lines
1.1 KiB
ReStructuredText
mindspore.ops.ScatterNdMax
|
||
===========================
|
||
|
||
.. py:class:: mindspore.ops.ScatterNdMax(use_locking=False)
|
||
|
||
对张量中的单个值或切片计算稀疏最大值。
|
||
|
||
使用给定值通过最大值运算和输入索引更新Parameter或Tensor值。在更新完成后输出 `input_x` ，这有利于更加方便地使用更新后的值。
|
||
|
||
更多参考详见 :func:`mindspore.ops.scatter_nd_max` 。
|
||
|
||
参数：
|
||
- **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` 相同。
|