CollectivePermute

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

Операция для перестановки тензоров в реплицируемых экземплярах TPU.

Каждый экземпляр предоставляет свои собственные входные данные.

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

Константы

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

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

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

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

Константы

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

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

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

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

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

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

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

public static CollectivePermute <T> create (область действия , ввод операнда <T>, операнд <TInt32> sourceTargetPairs)

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

Параметры
объем текущий объем
вход Локальный ввод для перестановки. В настоящее время поддерживаются только float и bfloat16.
источникTargetPairs Тензор формы [num_pairs, 2].
Возврат
  • новый экземпляр CollectivePermute

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

Перестановочный ввод.