AllToAll

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

Операция для обмена данными между репликами TPU.

В каждой реплике входные данные разбиваются на блоки «split_count» по «split_dimension» и отправляются другим репликам с заданным параметром group_assignment. После получения блоков «split_count» — 1 от других реплик мы объединяем блоки по «concat_dimension» в качестве вывода.

Например, предположим, что имеется 2 реплики TPU: реплика 0 получает входные данные: `[[A, B]]` реплика 1 получает входные данные: `[[C, D]]`

group_assignment=`[[0, 1]]` concat_dimension=0 Split_dimension=1 Split_count=2

Вывод реплики 0: `[[A], [C]]` Вывод реплики 1: `[[B], [D]]`

Константы

Нить

OP_NAME

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

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

Выход <Т>	какВывод () Возвращает символический дескриптор тензора.
статический <T расширяет TType > AllToAll <T>	create ( Область действия, ввод операнда <T>, операнд <TInt32> groupAssignment, Long concatDimension, Long SplitDimension, Long SplitCount) Фабричный метод для создания класса, обертывающего новую операцию AllToAll.
Выход <Т>	выход () Обмененный результат.

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

Из класса org.tensorflow.op.RawOp

последнее логическое значение	равно (Объект obj)
окончательный интервал	хэш-код ()
Операция	оп () Верните эту единицу вычислений как одну Operation .
последняя строка	нанизывать ()

Из класса java.lang.Object

логическое значение	равно (Объект arg0)
последний класс<?>	получитьКласс ()
интервал	хэш-код ()
окончательная пустота	поставить в известность ()
окончательная пустота	уведомитьВсе ()
Нить	нанизывать ()
окончательная пустота	подождать (длинный arg0, int arg1)
окончательная пустота	подождите (длинный arg0)
окончательная пустота	ждать ()

Из интерфейса org.tensorflow.op.Op

абстрактная среда выполнения	окружение () Верните среду выполнения, в которой была создана эта операция.
абстрактная операция	оп () Верните эту единицу вычислений как одну Operation .

Из интерфейса org.tensorflow.Operand

абстрактный вывод <T>	какВывод () Возвращает символический дескриптор тензора.
абстрактная Т	асТензор () Возвращает тензор в этом операнде.
абстрактная форма	форма () Возвращает (возможно, частично известную) форму тензора, на который ссылается Output этого операнда.
абстрактный класс<T>	тип () Возвращает тензорный тип этого операнда

Из интерфейса org.tensorflow.ndarray.Shaped

абстрактный int	классифицировать ()
абстрактная форма	форма ()
абстрактный длинный	размер () Вычисляет и возвращает общий размер этого контейнера в количестве значений.

Константы

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

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

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

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

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

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

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

public static AllToAll <T> create (область действия , ввод операнда <T>, операнд <TInt32> groupAssignment, Long concatDimension, Long SplitDimension, Long SplitCount)

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

Параметры

объем	текущий объем
вход	Локальный вход в сумму.
группаНазначение	Тензор int32 формы [num_groups, num_replicas_per_group]. `group_assignment[i]` представляет идентификаторы реплик в i-й подгруппе.
concatDimension	Число измерения для объединения.
SplitDimension	Число измерения, которое нужно разделить.
СплитКаунт	Количество разбиений, это число должно равняться размеру подгруппы(group_assignment.get_shape()[1])

Возврат

• новый экземпляр AllToAll

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

Обмененный результат.