tf.tensor_scatter_nd_max
Применяет разреженное обновление к тензору, выбирая элемент-мудрое максимальное значение.
tf.tensor_scatter_nd_max(
tensor, indices, updates, name=None
)
Возвращает новый тензор, скопированный из tensor, значения которого являются элемент-мудро максимальными между тензором и обновлениями в соответствии с индексами.
tensor = [0, 0, 0, 0, 0, 0, 0, 0] indices = [[1], [4], [5]] updates = [1, -1, 1] tf.tensor_scatter_nd_max(tensor, indices, updates).numpy() array([0, 1, 0, 0, 0, 1, 0, 0], dtype=int32)
Обратитесь к tf.tensor_scatter_nd_update для получения более подробной информации.
| Аргументы | |
|---|---|
tensor | A Tensor. Обновляемый тензор. |
indices | A Tensor. Должен быть одного из следующих типов: int32, int64. Индексный тензор. |
updates | A Tensor. Должен иметь тот же тип, что и tensor. Обновления для внедрения в вывод. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет тот же тип, что и tensor. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/tensor_scatter_nd_max