Вычисляет градиенты для SparseSegmentSum.
Возвращает «выходные данные» тензора той же формы, что и град, за исключением измерения 0, значением которого является количество уникальных индексов в «индексах». Также возвращает вектор sorted_unique_indices, содержащий соответствующие индексы из «indices».
Публичные методы
static <T расширяет число, U расширяет число, V расширяет число> SparseSegmentSumGradV2 <T, U> | |
Выход <Т> | выход () |
Выход <U> |
Унаследованные методы
Публичные методы
public static SparseSegmentSumGradV2 <T, U> create (область области действия , операнд <T> град, операнд <U> индексы, операнд <V> сегментные идентификаторы, операнд <Integer> DensityOutputDim0)
Фабричный метод для создания класса, обертывающего новую операцию SparseSegmentSumGradV2.
Параметры
объем | текущий объем |
---|---|
выпускник | градиент передается в операцию SparseSegmentSum. |
индексы | индексы, передаваемые в соответствующую операцию SparseSegmentSum. |
идентификаторы сегментов | Segment_ids передается в соответствующую операцию SparseSegmentSum. |
плотныйВыходДим0 | измерение 0 «данных», переданных в SparseSegmentSum op. |
Возврат
- новый экземпляр SparseSegmentSumGradV2