CollectiveGatherV2

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

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

`is_stateless` означает, что каждая операция не нуждается в зависимостях управления от других коллективных операций. В этом случае для различения коллективных групп следует использовать ключи, уникальные во время выполнения (например, «instance_key»).

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

сорт CollectiveGatherV2.Параметры Дополнительные атрибуты для CollectiveGatherV2

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

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

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

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

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

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

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

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

public static CollectiveGatherV2 <T> create ( Область действия, ввод операнда <T>, операнд <Integer> groupSize, операнд <Integer> groupKey, операнд <Integer> экземплярный ключ, Iterable< операнд <?>> orderingToken, параметры... параметры)

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

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

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

общедоступный статический CollectiveGatherV2.Options isStateless (Boolean isStateless)

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