SparseSegmentSumGradV2

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

Вычисляет градиенты для SparseSegmentSum.

Возвращает «выходные данные» тензора той же формы, что и град, за исключением измерения 0, значением которого является количество уникальных индексов в «индексах». Также возвращает вектор sorted_unique_indices, содержащий соответствующие индексы из «indices».

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

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

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

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

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

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

Параметры
объем текущий объем
выпускник градиент передается в операцию SparseSegmentSum.
индексы индексы, передаваемые в соответствующую операцию SparseSegmentSum.
идентификаторы сегментов Segment_ids передается в соответствующую операцию SparseSegmentSum.
плотныйВыходДим0 измерение 0 «данных», переданных в SparseSegmentSum op.
Возврат
  • новый экземпляр SparseSegmentSumGradV2

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

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