CollectiveGather

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

Взаимно накапливает несколько тензоров одинакового типа и формы.

Вложенные классы

сорт CollectiveGather.Options Дополнительные атрибуты для CollectiveGather

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

Выход <Т>
какВывод ()
Возвращает символический дескриптор тензора.
статический CollectiveGather.Options
communicationHint (Строка communicationHint)
static <T расширяет число> CollectiveGather <T>
create ( Область действия , ввод операнда <T>, длинный groupSize, длинный groupKey, длинный экземплярKey, форма фигуры , параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию CollectiveGather.
Выход <Т>
статический CollectiveGather.Options
timeoutSeconds (Плавающее значение timeoutSeconds)

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

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

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

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

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

общедоступный статический CollectiveGather.Options communicationHint (String communicationHint)

public static CollectiveGather <T> create (область области действия , ввод операнда <T>, длинный groupSize, длинный groupKey, длинный экземплярKey, форма формы , параметры... параметры)

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

Параметры
объем текущий объем
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр CollectiveGather

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

общедоступный статический CollectiveGather.Options timeoutSeconds (Float timeoutSeconds)