tf.compat.v1.scatter_max

Сводит разреженные обновления к ссылке на переменную с помощью операции max.

tf.compat.v1.scatter_max(
    ref, indices, updates, use_locking=False, name=None
)

Эта операция вычисляет

# Scalar indices
ref[indices, ...] = max(ref[indices, ...], updates[...])

# Vector indices (for each i)
ref[indices[i], ...] = max(ref[indices[i], ...], updates[i, ...])

# High rank indices (for each i, ..., j)
ref[indices[i, ..., j], ...] = max(ref[indices[i, ..., j], ...],
updates[i, ..., j, ...])

После завершения обновления операция возвращает ref. Это упрощает объединение в цепочку операций, которым требуется использовать сброшенное значение.

Повторяющиеся записи обрабатываются корректно: если несколько indices ссылаются на одно и то же расположение, их вклады объединяются.

Требуется updates.shape = indices.shape + ref.shape[1:] или updates.shape = [].

Аргументы
`ref`	Изменяемый `Tensor`. Допустимые типы: `half`, `bfloat16`, `float32`, `float64`, `int32`, `int64`. Должен быть получен из узла `Variable`.
`indices`	`Tensor`. Допустимые типы: `int32`, `int64`. Тензор индексов в первое измерение `ref`.
`updates`	`Tensor`. Должен иметь тот же тип, что и `ref`. Тензор обновлённых значений, которые требуется свести к `ref`.
`use_locking`	Необязательный `bool`. По умолчанию — `False`. Если значение True, обновление будет защищено блокировкой; в противном случае поведение не определено, но может сопровождаться меньшей конкуренцией за ресурсы.
`name`	Имя операции (необязательно).

Возвращает
Изменяемый `Tensor`. Имеет тот же тип, что и `ref`.

© 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/api_docs/python/tf/compat/v1/scatter_max