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