AllToAll

الفئة النهائية العامة AllToAll

عملية لتبادل البيانات عبر نسخ TPU المتماثلة.

في كل نسخة متماثلة، يتم تقسيم الإدخال إلى كتل `split_count` بطول `split_dimension` وإرسالها إلى النسخ المتماثلة الأخرى المعطاة لمجموعة_تعيين. بعد تلقي كتل `split_count` - 1 من النسخ المتماثلة الأخرى، نقوم بتسلسل الكتل على طول `concat_dimension` كمخرجات.

على سبيل المثال، لنفترض أن هناك نسختين متماثلتين من مادة TPU: النسخة المتماثلة 0 تتلقى الإدخال: `[[A, B]]` النسخة المتماثلة 1 تتلقى الإدخال: `[[C, D]]`

group_signment=`[[0, 1]]` concat_dimension=0 Split_dimension=1 Split_count=2

إخراج النسخة المتماثلة 0: `[[A], [C]]` إخراج النسخة المتماثلة 1: `[[B], [D]]`

الثوابت

خيط OP_NAME اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي

الأساليب العامة

الإخراج <T>
كإخراج ()
إرجاع المقبض الرمزي للموتر.
ثابت <T يمتد TType > AllToAll <T>
إنشاء (نطاق النطاق ، إدخال المعامل <T>، المعامل < TInt32 > groupAssignment، Long concatDimension، Long SplitDimension، Long SplitCount)
طريقة المصنع لإنشاء فئة تغلف عملية AllToAll جديدة.
الإخراج <T>
انتاج ()
النتيجة المتبادلة.

الطرق الموروثة

org.tensorflow.op.RawOp
منطقية نهائية
يساوي (كائن كائن)
كثافة العمليات النهائية
رمز التجزئة ()
عملية
المرجع ()
قم بإرجاع وحدة الحساب هذه Operation واحدة.
السلسلة النهائية
إلى سلسلة ()
منطقية
يساوي (كائن arg0)
الدرجة النهائية<?>
الحصول على كلاس ()
كثافة العمليات
رمز التجزئة ()
الفراغ النهائي
إعلام ()
الفراغ النهائي
إعلام الكل ()
خيط
إلى سلسلة ()
الفراغ النهائي
انتظر (طويل arg0، int arg1)
الفراغ النهائي
انتظر (طويل arg0)
الفراغ النهائي
انتظر ()
org.tensorflow.op.Op
بيئة التنفيذ المجردة
بيئى ()
قم بإرجاع بيئة التنفيذ التي تم إنشاء هذه العملية فيها.
عملية مجردة
المرجع ()
قم بإرجاع وحدة الحساب هذه Operation واحدة.
org.tensorflow.Operand
إخراج مجردة <T>
كإخراج ()
إرجاع المقبض الرمزي للموتر.
مجردة T
الموتر ()
إرجاع الموتر في هذا المعامل.
الشكل المجرد
شكل ()
تُرجع الشكل (ربما المعروف جزئيًا) للموتر المشار إليه في Output هذا المعامل.
فئة مجردة<T>
يكتب ()
إرجاع نوع الموتر لهذا المعامل
org.tensorflow.ndarray.Shaped
كثافة العمليات مجردة
رتبة ()
الشكل المجرد
شكل ()
مجردة طويلة
مقاس ()
يحسب ويعيد الحجم الإجمالي لهذه الحاوية بعدد القيم.

الثوابت

السلسلة النهائية الثابتة العامة OP_NAME

اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي

القيمة الثابتة: "AllToAll"

الأساليب العامة

الإخراج العام <T> كإخراج ()

إرجاع المقبض الرمزي للموتر.

المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.

إنشاء AllToAll <T> ثابت عام (نطاق النطاق ، إدخال المعامل <T>، المعامل < TInt32 > groupAssignment، Long concatDimension، Long SplitDimension، Long SplitCount)

طريقة المصنع لإنشاء فئة تغلف عملية AllToAll جديدة.

حدود
نِطَاق النطاق الحالي
مدخل الإدخال المحلي للمجموع.
مهمه المجموعه موتر int32 بالشكل [num_groups, num_replicas_per_group]. يمثل `group_signment[i]` المعرفات المتماثلة في المجموعة الفرعية i.
concatDimension رقم البعد المراد تسلسله.
com.splitDimension رقم البعد المراد تقسيمه.
SplitCount عدد الانقسامات، يجب أن يكون هذا العدد مساويا لحجم المجموعة الفرعية (group_assiment.get_shape()[1])
عائدات
  • مثيل جديد من AllToAll

الإخراج العام <T> الإخراج ()

النتيجة المتبادلة.