tf.keras.ops.segment_max

Вычисляет максимальные значения сегментов в тензоре.

tf.keras.ops.segment_max(
    data, segment_ids, num_segments=None, sorted=False
)

Аргументы
`data`	Входной тензор.
`segment_ids`	Одномерный тензор, содержащий индексы сегментов для каждого элемента в `data`.
`num_segments`	Целое число, обозначающее общее количество сегментов. Если не указано, оно определяется по максимальному значению в `segment_ids`.
`sorted`	Логическое значение, указывающее, отсортирован ли `segment_ids`. По умолчанию — `False`.

Возвращает
Тензор, содержащий максимальные значения сегментов, где каждый элемент представляет максимальное значение соответствующего сегмента в `data`.

Пример:

data = keras.ops.convert_to_tensor([1, 2, 10, 20, 100, 200])
segment_ids = keras.ops.convert_to_tensor([0, 0, 1, 1, 2, 2])
num_segments = 3
keras.ops.segment_max(data, segment_ids, num_segments)
array([2, 20, 200], dtype=int32)

© 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/keras/ops/segment_max