NcclAllReduce

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

Выводит тензор, содержащий сокращение всех входных тензоров.

Выводит тензор, содержащий сокращение всех входных тензоров, переданных в операции с одним и тем же `shared_name.

Граф должен быть построен таким образом, чтобы если одна операция выполняется со значением общего_имени `c`, то операции `num_devices` будут выполняться со значением общего_имени `c`. Если этого не сделать, выполнение графа не завершится.

ввод: ввод данных сокращения: значение сокращения для всех устройств `num_devices`. сокращение: операция сокращения, которую необходимо выполнить. num_devices: количество устройств, участвующих в этом сокращении. shared_name: Идентификатор, который используется совместно операциями одного и того же сокращения.

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

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

Выход <Т>
какВывод ()
Возвращает символический дескриптор тензора.
статический <T расширяет TNumber > NcclAllReduce <T>
create (область действия , ввод операнда <T>, сокращение строки, длинное число устройств, строка общего имени)
Фабричный метод для создания класса, обертывающего новую операцию NcclAllReduce.
Выход <Т>

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

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «NcclAllReduce».

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

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

Возвращает символический дескриптор тензора.

Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.

public static NcclAllReduce <T> create (область действия , ввод операнда <T>, сокращение строки, длинное число устройств, строка общего имени)

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

Параметры
объем текущий объем
Возврат
  • новый экземпляр NcclAllReduce

общедоступные выходные данные <T> ()