SegmentSumV2

публичный финальный класс SegmentSumV2

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

Прочтите [раздел о сегментации](https://tensorflow.org/api_docs/python/tf/math#Segmentation) для объяснения сегментов.

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

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

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

Публичные методы

Выход <Т>
какВывод ()
Возвращает символический дескриптор тензора.
static <T, U расширяет число, V расширяет число> SegmentSumV2 <T>
create (Область действия , данные операнда <T>, идентификаторы сегментов операнда <U>, число сегментов операнда <V>)
Фабричный метод для создания класса, обертывающего новую операцию SegmentSumV2.
Выход <Т>
выход ()
Имеет ту же форму, что и данные, за исключением первых измерений «segment_ids.rank», которые заменяются одним измерением с размером «num_segments».

Унаследованные методы

Публичные методы

публичный вывод <T> asOutput ()

Возвращает символический дескриптор тензора.

Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.

public static SegmentSumV2 <T> create (область действия , данные операнда <T>, идентификаторы сегментов операнда <U>, операнд <V> numSegments)

Фабричный метод для создания класса, обертывающего новую операцию SegmentSumV2.

Параметры
объем текущий объем
идентификаторы сегментов Одномерный тензор, размер которого равен размеру первого измерения данных. Значения должны быть отсортированы и могут повторяться. Значения должны быть меньше `num_segments`.

Внимание: значения всегда проверяются для сортировки на ЦП и никогда не проверяются на графическом процессоре.

Возврат
  • новый экземпляр SegmentSumV2

публичный вывод <T> вывод ()

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