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) |
| الفراغ النهائي | انتظر () |
الثوابت
السلسلة النهائية الثابتة العامة OP_NAME
اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي
القيمة الثابتة: "CollectivePermute"
الأساليب العامة
الإخراج العام <T> كإخراج ()
إرجاع المقبض الرمزي للموتر.
المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.
طريقة المصنع لإنشاء فئة تغلف عملية CollectivePermute جديدة.
حدود
| نِطَاق | النطاق الحالي |
|---|
| مدخل | المدخلات المحلية المراد تبديلها. حاليًا يدعم فقط float وbfloat16. |
|---|
| sourceTargetPairs | موتر ذو شكل [num_pairs, 2]. |
|---|
عائدات
- مثيل جديد من CollectivePermute
إنّ محتوى هذه الصفحة مرخّص بموجب ترخيص Creative Commons Attribution 4.0 ما لم يُنصّ على خلاف ذلك، ونماذج الرموز مرخّصة بموجب ترخيص Apache 2.0. للاطّلاع على التفاصيل، يُرجى مراجعة سياسات موقع Google Developers. إنّ Java هي علامة تجارية مسجَّلة لشركة Oracle و/أو شركائها التابعين.
تاريخ التعديل الأخير: 2025-07-26 (حسب التوقيت العالمي المتفَّق عليه)
[null,null,["تاريخ التعديل الأخير: 2025-07-26 (حسب التوقيت العالمي المتفَّق عليه)"],[],[],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."]]
