tf.raw_ops.UnsortedSegmentMax

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

Просмотреть псевдонимы

Псевдонимы для совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.raw_ops.UnsortedSegmentMax

tf.raw_ops.UnsortedSegmentMax(
    data, segment_ids, num_segments, name=None
)

Объяснение сегментов см. в разделе о сегментации.

Этот оператор похож на tf.math.unsorted_segment_sum. Вместо вычисления суммы по сегментам он вычисляет максимум, так что:

\(output_i = \max_{j...} data[j...]\), где максимум берётся по кортежам j..., таким что segment_ids[j...] == i.

Если для заданного идентификатора сегмента i максимум не определён, оператор возвращает наименьшее возможное значение для указанного числового типа, output[i] = numeric_limits<T>::lowest().

Если заданный идентификатор сегмента i отрицателен, соответствующее значение отбрасывается и не включается в результат.

Например:

c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.math.unsorted_segment_max(c, tf.constant([0, 1, 0]), num_segments=2).numpy()
array([[4, 3, 3, 4],
       [5,  6, 7, 8]], dtype=int32)

Аргументы
`data`	`Tensor`. Допустимые типы: `float32`, `float64`, `int32`, `uint8`, `int16`, `int8`, `int64`, `bfloat16`, `uint16`, `half`, `uint32`, `uint64`.
`segment_ids`	`Tensor`. Допустимые типы: `int32`, `int64`. Тензор, форма которого является префиксом `data.shape`. Значения должны быть меньше `num_segments`. Внимание: На CPU значения всегда проверяются на принадлежность диапазону, на GPU проверка не выполняется.
`num_segments`	`Tensor`. Допустимые типы: `int32`, `int64`.
`name`	Имя операции (необязательно).

Возвращает
`Tensor`. Имеет тот же тип, что и `data`.

© 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/raw_ops/UnsortedSegmentMax