публичный финальный класс AllReduce
Взаимно уменьшает несколько тензоров одинакового типа и формы.
Вложенные классы
| сорт | AllReduce.Параметры | Дополнительные атрибуты для AllReduce | |
Константы
| Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
| Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
| статический AllReduce.Options | communicationHint (Строка communicationHint) |
| статический <T расширяет TNumber > AllReduce <T> | create ( Область области, ввод операнда <T>, длинный groupSize, длинный groupKey, длинный экземплярKey, String mergeOp, String FinalOp, List<Long> subdivOffsets, Options... options) Фабричный метод для создания класса, обертывающего новую операцию AllReduce. |
| Выход <Т> | данные () |
| статический AllReduce.Options | timeoutSeconds (Плавающее значение timeoutSeconds) |
| статический AllReduce.Options | waitFor (Список<Long> waitFor) |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Постоянное значение: «CollectiveReduce».
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static AllReduce <T> create (область области действия , ввод операнда <T>, длинный groupSize, длинный groupKey, длинный экземплярKey, String mergeOp, String FinalOp, List<Long> subdivOffsets, Options... options)
Фабричный метод для создания класса, обертывающего новую операцию AllReduce.
Параметры
| объем | текущий объем |
|---|---|
| параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр AllReduce