Уменьшает «вход» из «num_devices», используя «сокращение» до одного устройства.
Уменьшает «вход» из «num_devices», используя «сокращение» до одного устройства.
Граф должен быть построен так, чтобы все входы имели допустимое назначение устройств, а самому оператору было назначено одно из этих устройств.
вход: вход для сокращения. данные: значение сокращения для всех устройств `num_devices`. сокращение: операция сокращения, которую необходимо выполнить.
Публичные методы
| Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
| статический <T расширяет число> NcclReduce <T> | create (область области действия , ввод Iterable< Operand <T>>, сокращение строк) Фабричный метод для создания класса, обертывающего новую операцию NcclReduce. |
| Выход <Т> | данные () |
Унаследованные методы
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static NcclReduce <T> create (область области действия , ввод Iterable< Operand <T>>, сокращение строк)
Фабричный метод для создания класса, обертывающего новую операцию NcclReduce.
Параметры
| объем | текущий объем |
|---|
Возврат
- новый экземпляр NcclReduce