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