SparseSegmentMeanGradV2

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

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

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

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

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

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

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

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

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

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

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

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