публичный финальный класс CollectiveAllToAllV2
Взаимно меняет несколько тензоров одинакового типа и формы.
`is_stateless` означает, что каждая операция не нуждается в зависимостях управления от других коллективных операций. В этом случае для различения коллективных групп следует использовать ключи, уникальные во время выполнения (например, «instance_key»).
Вложенные классы
| сорт | КоллективAllToAllV2.Options | Дополнительные атрибуты для CollectiveAllToAllV2 | |
Публичные методы
| Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
| статический CollectiveAllToAllV2.Options | communicationHint (Строка communicationHint) |
| static <T расширяет номер> CollectiveAllToAllV2 <T> | |
| Выход <Т> | данные () |
| статический CollectiveAllToAllV2.Options | isStateless (логическое значение isStateless) |
| статический CollectiveAllToAllV2.Options | timeoutSeconds (Плавающее значение timeoutSeconds) |
Унаследованные методы
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static CollectiveAllToAllV2 <T> create ( Область действия, ввод операнда <T>, операнд <Integer> groupSize, операнд <Integer> groupKey, операнд <Integer> instanceKey, Iterable< операнд <?>> orderingToken, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию CollectiveAllToAllV2.
Параметры
| объем | текущий объем |
|---|---|
| параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр CollectiveAllToAllV2