NcclBroadcast

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

Отправляет вход всем устройствам, подключенным к выходу.

Отправляет вход всем устройствам, подключенным к выходу.

Граф должен быть построен так, чтобы все операции, подключенные к выходу, имели допустимое назначение устройства, а самому оператору было назначено одно из этих устройств.

вход: вход для трансляции. вывод: То же, что и ввод. shape: Форма входного тензора.

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

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

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

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

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

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

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

public static NcclBroadcast <T> create (область области действия , ввод операнда <T>, форма формы )

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

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

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