NcclAllReduce

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

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

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

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

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

Константы

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

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

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

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

org.tensorflow.op.RawOp
последнее логическое значение
равно (Объект obj)
окончательный интервал
хэш-код ()
Операция
оп ()
Верните эту единицу вычислений как одну Operation .
последняя строка
нанизывать ()
логическое значение
равно (Объект arg0)
последний класс<?>
получитьКласс ()
интервал
хэш-код ()
окончательная пустота
поставить в известность ()
окончательная пустота
уведомитьВсе ()
Нить
нанизывать ()
окончательная пустота
подождать (длинный arg0, int arg1)
окончательная пустота
подождите (длинный arg0)
окончательная пустота
ждать ()
org.tensorflow.op.Op
абстрактная среда выполнения
окружение ()
Верните среду выполнения, в которой была создана эта операция.
абстрактная операция
оп ()
Верните эту единицу вычислений как одну Operation .
org.tensorflow.Operand
абстрактный вывод <T>
какВывод ()
Возвращает символический дескриптор тензора.
абстрактная Т
асТензор ()
Возвращает тензор в этом операнде.
абстрактная форма
форма ()
Возвращает (возможно, частично известную) форму тензора, на который ссылается Output этого операнда.
абстрактный класс<T>
тип ()
Возвращает тензорный тип этого операнда
org.tensorflow.ndarray.Shaped
абстрактный int
классифицировать ()
абстрактная форма
форма ()
абстрактный длинный
размер ()
Вычисляет и возвращает общий размер этого контейнера в количестве значений.

Константы

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

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

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

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

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

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

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

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

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

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

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