Вычисляет сумму по сегментам тензора.
Прочтите [раздел о сегментации](https://tensorflow.org/api_docs/python/tf/math#Segmentation) для объяснения сегментов.
Вычисляет тензор такой, что \\(output[i] = \sum_{j...} data[j...]\\) где сумма вычисляется по кортежам `j...` таким, что `segment_ids[j...] == i`. В отличие от SegmentSum, Segment_ids не нужно сортировать и не обязательно охватывать все значения во всем диапазоне допустимых значений.
Если сумма пуста для данного идентификатора сегмента `i`, `output[i] = 0`. Если заданный идентификатор сегмента «i» отрицательный, значение отбрасывается и не будет добавлено к сумме сегмента.
`num_segments` должно равняться количеству различных идентификаторов сегментов.
c = tf.constant([[1,2,3,4], [5,6,7,8], [4,3,2,1]])
tf.unsorted_segment_sum(c, tf.constant([0, 1, 0]), num_segments=2)
# ==> [[ 5, 5, 5, 5],
# [5, 6, 7, 8]]
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
static <T расширяет TType > UnsortedSegmentSum <T> | |
Выход <Т> | выход () Имеет ту же форму, что и данные, за исключением первых измерений «segment_ids.rank», которые заменяются одним измерением с размером «num_segments». |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static UnsortedSegmentSum <T> create (область действия , данные операнда <T>, операнд <? расширяет TNumber > сегменты, операнд <? расширяет TNumber > numSegments)
Фабричный метод для создания класса, обертывающего новую операцию UnsortedSegmentSum.
Параметры
объем | текущий объем |
---|---|
идентификаторы сегментов | Тензор, форма которого является префиксом data.shape. |
Возврат
- новый экземпляр UnsortedSegmentSum
публичный вывод <T> вывод ()
Имеет ту же форму, что и данные, за исключением первых измерений «segment_ids.rank», которые заменяются одним измерением с размером «num_segments».