Spec-Zone.ru › TensorFlow

tf.raw_ops.SegmentMin

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

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

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

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.raw_ops.SegmentMin

tf.raw_ops.SegmentMin(
    data, segment_ids, name=None
)

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

Вычисляет тензор, для которого \(output_i = \min_j(data_j)\), где min выполняется по j, таким что segment_ids[j] == i.

Если для заданного идентификатора сегмента i минимум отсутствует, output[i] = 0.

Внимание: На CPU значения в segment_ids всегда проверяются на сортировку, и для индексов, которые не расположены в возрастающем порядке, возникает ошибка. На GPU ошибка для несортированных индексов не возникает. На GPU индексы, расположенные не по порядку, приводят к безопасному, но неопределённому поведению. В частности, такие индексы могут обрабатываться как совпадающие с меньшим следующим индексом.

Например:

c = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]])
tf.math.segment_min(c, tf.constant([0, 0, 1])).numpy()
array([[1, 2, 2, 1],
       [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. Значения должны быть отсортированы; повторения допускаются.
Внимание: На CPU значения всегда проверяются на сортировку, на GPU проверка не выполняется.
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/SegmentMin

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API