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. |
Унаследованные методы
Из класса java.lang.Object | логическое значение | равно (Объект arg0) |
| последний класс<?> | получитьКласс () |
| интервал | хэш-код () |
| окончательная пустота | поставить в известность () |
| окончательная пустота | уведомитьВсе () |
| Нить | нанизывать () |
| окончательная пустота | подождать (длинный arg0, int arg1) |
| окончательная пустота | подождите (длинный arg0) |
| окончательная пустота | ждать () |
Из интерфейса org.tensorflow.Operand | абстрактный вывод <T> | какВывод () Возвращает символический дескриптор тензора. |
| абстрактная Т | асТензор () Возвращает тензор в этом операнде. |
| абстрактная форма | форма () Возвращает (возможно, частично известную) форму тензора, на который ссылается Output этого операнда. |
| абстрактный класс<T> | тип () Возвращает тензорный тип этого операнда |
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Постоянное значение: «CollectivePermute».
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
Фабричный метод для создания класса, обертывающего новую операцию CollectivePermute.
Параметры
| объем | текущий объем |
|---|
| вход | Локальный ввод для перестановки. В настоящее время поддерживаются только float и bfloat16. |
|---|
| источникTargetPairs | Тензор формы [num_pairs, 2]. |
|---|
Возврат
- новый экземпляр CollectivePermute
публичный вывод <T> вывод ()
Если не указано иное, контент на этой странице предоставляется по лицензии Creative Commons "С указанием авторства 4.0", а примеры кода – по лицензии Apache 2.0. Подробнее об этом написано в правилах сайта. Java – это зарегистрированный товарный знак корпорации Oracle и ее аффилированных лиц.
Последнее обновление: 2025-07-26 UTC.
[null,null,["Последнее обновление: 2025-07-26 UTC."],[],[],null,["# CollectivePermute\n\npublic final class **CollectivePermute** \nAn Op to permute tensors across replicated TPU instances.\n\n\nEach instance supplies its own input.\n\n\nFor example, suppose there are 4 TPU instances: \\`\\[A, B, C, D\\]\\`. Passing\nsource_target_pairs=\\`\\[\\[0,1\\],\\[1,2\\],\\[2,3\\],\\[3,0\\]\\]\\` gets the outputs:\n\\`\\[D, A, B, C\\]\\`.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n### Constants\n\n|--------|-------------------------------------------------------------------------------|---------------------------------------------------------|\n| String | [OP_NAME](/jvm/api_docs/java/org/tensorflow/op/tpu/CollectivePermute#OP_NAME) | The name of this op, as known by TensorFlow core engine |\n\n### Public Methods\n\n|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e | [asOutput](/jvm/api_docs/java/org/tensorflow/op/tpu/CollectivePermute#asOutput())() Returns the symbolic handle of the tensor. |\n| static \\\u003cT extends [TType](/jvm/api_docs/java/org/tensorflow/types/family/TType)\\\u003e [CollectivePermute](/jvm/api_docs/java/org/tensorflow/op/tpu/CollectivePermute)\\\u003cT\\\u003e | [create](/jvm/api_docs/java/org/tensorflow/op/tpu/CollectivePermute#create(org.tensorflow.op.Scope, org.tensorflow.Operand\u003cT\u003e, org.tensorflow.Operand\u003corg.tensorflow.types.TInt32\u003e))([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e input, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e sourceTargetPairs) Factory method to create a class wrapping a new CollectivePermute operation. |\n| [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e | [output](/jvm/api_docs/java/org/tensorflow/op/tpu/CollectivePermute#output())() The permuted input. |\n\n### Inherited Methods\n\nFrom class [org.tensorflow.op.RawOp](/jvm/api_docs/java/org/tensorflow/op/RawOp) \n\n|----------------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| final boolean | [equals](/jvm/api_docs/java/org/tensorflow/op/RawOp#equals(java.lang.Object))(Object obj) |\n| final int | [hashCode](/jvm/api_docs/java/org/tensorflow/op/RawOp#hashCode())() |\n| [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/RawOp#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n| final String | [toString](/jvm/api_docs/java/org/tensorflow/op/RawOp#toString())() |\n\nFrom class java.lang.Object \n\n|------------------|---------------------------|\n| boolean | equals(Object arg0) |\n| final Class\\\u003c?\\\u003e | getClass() |\n| int | hashCode() |\n| final void | notify() |\n| final void | notifyAll() |\n| String | toString() |\n| final void | wait(long arg0, int arg1) |\n| final void | wait(long arg0) |\n| final void | wait() |\n\nFrom interface [org.tensorflow.op.Op](/jvm/api_docs/java/org/tensorflow/op/Op) \n\n|-----------------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [ExecutionEnvironment](/jvm/api_docs/java/org/tensorflow/ExecutionEnvironment) | [env](/jvm/api_docs/java/org/tensorflow/op/Op#env())() Return the execution environment this op was created in. |\n| abstract [Operation](/jvm/api_docs/java/org/tensorflow/Operation) | [op](/jvm/api_docs/java/org/tensorflow/op/Op#op())() Return this unit of computation as a single [Operation](/jvm/api_docs/java/org/tensorflow/Operation). |\n\nFrom interface [org.tensorflow.Operand](/jvm/api_docs/java/org/tensorflow/Operand) \n\n|-------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e | [asOutput](/jvm/api_docs/java/org/tensorflow/Operand#asOutput())() Returns the symbolic handle of the tensor. |\n| abstract T | [asTensor](/jvm/api_docs/java/org/tensorflow/Operand#asTensor())() Returns the tensor at this operand. |\n| abstract [Shape](/jvm/api_docs/java/org/tensorflow/ndarray/Shape) | [shape](/jvm/api_docs/java/org/tensorflow/Operand#shape())() Returns the (possibly partially known) shape of the tensor referred to by the [Output](/jvm/api_docs/java/org/tensorflow/Output) of this operand. |\n| abstract Class\\\u003cT\\\u003e | [type](/jvm/api_docs/java/org/tensorflow/Operand#type())() Returns the tensor type of this operand |\n\nFrom interface [org.tensorflow.ndarray.Shaped](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped) \n\n|-------------------------------------------------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------|\n| abstract int | [rank](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped#rank())() |\n| abstract [Shape](/jvm/api_docs/java/org/tensorflow/ndarray/Shape) | [shape](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped#shape())() |\n| abstract long | [size](/jvm/api_docs/java/org/tensorflow/ndarray/Shaped#size())() Computes and returns the total size of this container, in number of values. |\n\nConstants\n---------\n\n#### public static final String\n**OP_NAME**\n\nThe name of this op, as known by TensorFlow core engine \nConstant Value: \"CollectivePermute\"\n\nPublic Methods\n--------------\n\n#### public [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e\n**asOutput**\n()\n\nReturns the symbolic handle of the tensor.\n\nInputs to TensorFlow operations are outputs of another TensorFlow operation. This method is\nused to obtain a symbolic handle that represents the computation of the input.\n\n\u003cbr /\u003e\n\n#### public static [CollectivePermute](/jvm/api_docs/java/org/tensorflow/op/tpu/CollectivePermute)\\\u003cT\\\u003e\n**create**\n([Scope](/jvm/api_docs/java/org/tensorflow/op/Scope) scope, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003cT\\\u003e input, [Operand](/jvm/api_docs/java/org/tensorflow/Operand)\\\u003c[TInt32](/jvm/api_docs/java/org/tensorflow/types/TInt32)\\\u003e sourceTargetPairs)\n\nFactory method to create a class wrapping a new CollectivePermute operation. \n\n##### Parameters\n\n| scope | current scope |\n| input | The local input to be permuted. Currently only supports float and bfloat16. |\n| sourceTargetPairs | A tensor with shape \\[num_pairs, 2\\]. |\n|-------------------|-----------------------------------------------------------------------------|\n\n##### Returns\n\n- a new instance of CollectivePermute \n\n#### public [Output](/jvm/api_docs/java/org/tensorflow/Output)\\\u003cT\\\u003e\n**output**\n()\n\nThe permuted input."]]
