tf.raw_ops.TensorScatterMax

Применяет разреженное обновление к тензору, вычисляя поэлементный максимум.

tf.raw_ops.TensorScatterMax(
    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 для получения дополнительной информации.