Spec-Zone.ru › TensorFlow

tf.math.segment_min

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

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

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

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

tf.compat.v1.math.segment_min, tf.compat.v1.segment_min

tf.math.segment_min(
    data: Annotated[Any, tf.raw_ops.Any],
    segment_ids: Annotated[Any, tf.raw_ops.Any],
    name=None
) -> Annotated[Any, tf.raw_ops.Any]

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

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

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

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

Например:

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 A Tensor. Должен быть одним из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64.
segment_ids A Tensor. Должен быть одним из следующих типов: int32, int64. 1-мерный тензор, размер которого равен размеру первого измерения data. Значения должны быть отсортированы и могут повторяться.
Внимание: Значения всегда проверяются на сортировку в ЦП, никогда не проверяются в видеокартах.
name Имя операции (необязательно).
Возвращает
A 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/math/segment_min

Spec-Zone.ru

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