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