تحذير: تم إهمال واجهة برمجة التطبيقات هذه وستتم إزالتها في إصدار مستقبلي من TensorFlow بعد استقرار الاستبدال .

AllToAll

الجمهور AllToAll الدرجة النهائية

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

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

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

group_assignment = `[[0، 1]]` concat_dimension = 0 split_dimension = 1 split_count = 2

إخراج النسخة المتماثلة 0: `[[A]، [C]]` نسخة طبق الأصل 1's output: `[[B]، [D]]`

الطرق العامة

الانتاج <T>
asOutput ()
إرجاع المقبض الرمزي للموتر.
ثابت <T> AllToAll <T>
إنشاء ( نطاق نطاق، المعامل <T> المدخلات، و المعامل <صحيح> groupAssignment، طويل concatDimension، طويل splitDimension، طويل splitCount)
طريقة المصنع لإنشاء فئة التفاف لعملية AllToAll جديدة.
الانتاج <T>
الانتاج ()
النتيجة المتبادلة.

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

الطرق العامة

العام الناتج <T> asOutput ()

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

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

ساكنة العام AllToAll <T> إنشاء ( نطاق نطاق، المعامل <T> المدخلات، و المعامل <صحيح> groupAssignment، طويل concatDimension، طويل splitDimension، طويل splitCount)

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

العوامل
مجال النطاق الحالي
إدخال المدخلات المحلية للمبلغ.
مهمه المجموعه موتر int32 مع الشكل [عدد_مجموعات ، عدد_النسخ_المجموعات_المتكررة]. يمثل "group_assignment [i]" معرفات النسخ المتماثلة في المجموعة الفرعية ith.
concatDimension رقم البعد المراد ربطه.
تقسيم رقم البعد المراد تقسيمه.
سبليتكونت عدد الانقسامات ، يجب أن يساوي هذا الرقم حجم المجموعة الفرعية (group_assignment.get_shape () [1])
عائدات
  • مثيل جديد من AllToAll

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

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