tensorflow::ops::SegmentMinV2

#include <math_ops.h>

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

Сводка

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

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

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

Примечание: в настоящее время эта операция поддерживается только при jit_compile=True.

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

Единственное отличие от SegmentMin — дополнительный входной параметр num_segments. Он помогает вычислить форму результата во время компиляции. num_segments должен быть согласован с segment_ids. Например, Max(segment_ids) должен быть равен num_segments - 1 для одномерного segment_ids. При несогласованном num_segments операция всё равно выполняется. Единственное отличие состоит в том, что размер результата определяется значением num_segments независимо от размера segment_ids и data. Если num_segments меньше ожидаемого размера результата, последние элементы игнорируются; если num_segments больше ожидаемого размера результата, последние элементы получают максимально возможное значение для конкретного числового типа.

Например:

.function(jit_compile=True) ... def test(c): ... return tf.raw_ops.SegmentMinV2(data=c, segment_ids=tf.constant([0, 0, 1]), num_segments=2) c = tf.constant([[1,2,3,4], [4, 3, 2, 1], [5,6,7,8]]) test(c).numpy() array([[1, 2, 2, 1], [5, 6, 7, 8]], dtype=int32)

Аргументы:

• scope: объект Scope
• segment_ids: одномерный тензор, размер которого равен размеру первого измерения data. Значения должны быть отсортированы; допускаются повторяющиеся значения. Значения должны быть меньше num_segments.

Внимание: на CPU значения всегда проверяются на сортировку, на GPU такая проверка не выполняется.

Возвращает:

• Output: имеет ту же форму, что и data, за исключением первых segment_ids.rank измерений, которые заменяются одним измерением размера num_segments.

Общедоступные атрибуты

Operation operation

::tensorflow::Output output

Общедоступные функции

 SegmentMinV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input data,
  ::tensorflow::Input segment_ids,
  ::tensorflow::Input num_segments
)

::tensorflow::Node * node() const 

operator::tensorflow::Input() const 

operator::tensorflow::Output() const