CollectivePermute
Tetap teratur dengan koleksi
Simpan dan kategorikan konten berdasarkan preferensi Anda.
Opsi untuk mengubah posisi tensor di seluruh instance TPU yang direplikasi.
Setiap instance menyediakan masukannya sendiri.
Misalnya, ada 4 instance TPU: `[A, B, C, D]`. Melewati source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` akan menghasilkan keluaran: `[D, A, B, C]`.
Konstanta
| Rangkaian | OP_NAME | Nama operasi ini dikenal dengan mesin inti TensorFlow |
Metode Warisan
Dari kelas java.lang.Object | boolean | sama dengan (Objek arg0) |
| Kelas terakhir<?> | dapatkan Kelas () |
| ke dalam | Kode hash () |
| kekosongan terakhir | memberitahu () |
| kekosongan terakhir | beri tahuSemua () |
| Rangkaian | keString () |
| kekosongan terakhir | tunggu (arg0 panjang, int arg1) |
| kekosongan terakhir | tunggu (argumen panjang0) |
| kekosongan terakhir | Tunggu () |
Konstanta
String akhir statis publik OP_NAME
Nama operasi ini dikenal dengan mesin inti TensorFlow
Nilai Konstan: "CollectivePermute"
Metode Publik
Keluaran publik <T> sebagai Keluaran ()
Mengembalikan pegangan simbolis tensor.
Masukan ke operasi TensorFlow adalah keluaran dari operasi TensorFlow lainnya. Metode ini digunakan untuk mendapatkan pegangan simbolis yang mewakili perhitungan input.
Metode pabrik untuk membuat kelas yang membungkus operasi CollectivePermute baru.
Parameter
| cakupan | ruang lingkup saat ini |
|---|
| memasukkan | Masukan lokal yang akan diubah. Saat ini hanya mendukung float dan bfloat16. |
|---|
| sourceTargetPairs | Tensor dengan bentuk [num_pairs, 2]. |
|---|
Kembali
- contoh baru dari CollectivePermute
Kecuali dinyatakan lain, konten di halaman ini dilisensikan berdasarkan Lisensi Creative Commons Attribution 4.0, sedangkan contoh kode dilisensikan berdasarkan Lisensi Apache 2.0. Untuk mengetahui informasi selengkapnya, lihat Kebijakan Situs Google Developers. Java adalah merek dagang terdaftar dari Oracle dan/atau afiliasinya.
Terakhir diperbarui pada 2025-07-26 UTC.
[null,null,["Terakhir diperbarui pada 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."]]
