публичный финальный класс CollectivePermute
Операция для перестановки тензоров в реплицируемых экземплярах TPU.
Каждый экземпляр предоставляет свои собственные входные данные.
Например, предположим, что имеется 4 экземпляра TPU: `[A, B, C, D]`. Передача source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` получает выходные данные: `[D, A, B, C]`.
Публичные методы
| Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
| статический <T> CollectivePermute <T> | create ( Область действия , ввод операнда <T>, операнд <Integer> sourceTargetPairs) Фабричный метод для создания класса, обертывающего новую операцию CollectivePermute. |
| Выход <Т> | выход () Перестановочный ввод. |
Унаследованные методы
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static CollectivePermute <T> create (область действия , ввод операнда <T>, операнд <Integer> sourceTargetPairs)
Фабричный метод для создания класса, обертывающего новую операцию CollectivePermute.
Параметры
| объем | текущий объем |
|---|---|
| вход | Локальный ввод для перестановки. В настоящее время поддерживаются только float и bfloat16. |
| источникTargetPairs | Тензор формы [num_pairs, 2]. |
Возврат
- новый экземпляр CollectivePermute