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