tf.raw_ops.UnsortedSegmentMin
Вычисляет минимальное значение по сегментам тензора.
tf.raw_ops.UnsortedSegmentMin(
data, segment_ids, num_segments, name=None
)
Прочитайте раздел по сегментации для получения объяснения сегментов.
Этот оператор похож на оператор суммирования по несортированным сегментам, который можно найти (здесь). Вместо вычисления суммы по сегментам он вычисляет минимум так, что:
\(output_i = \min_{j...} data_[j...]\) где min берется по наборам j... таким образом, что segment_ids[j...] == i.
Если минимум пуст для данного идентификатора сегмента i, он выводит максимально возможное значение для конкретного числового типа, output[i] = numeric_limits<T>::max().
Например:
c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]]) tf.unsorted_segment_min(c, tf.constant([0, 1, 0]), num_segments=2) # ==> [[ 1, 2, 2, 1], # [5, 6, 7, 8]]
Если заданный идентификатор сегмента i отрицательный, соответствующее значение отбрасывается и не включается в результат.
| Аргументы | |
|---|---|
data | A Tensor. Должен быть одним из следующих типов: float32, float64, int32, uint8, int16, int8, int64, bfloat16, uint16, half, uint32, uint64. |
segment_ids | A Tensor. Должен быть одним из следующих типов: int32, int64. Тензор, форма которого является префиксом data.shape. |
num_segments | A Tensor. Должен быть одним из следующих типов: int32, int64. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor. Имеет тот же тип, что и data. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/UnsortedSegmentMin