الفئة النهائية العامة 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 الأساسي |
الأساليب العامة
الطرق الموروثة
| منطقية | يساوي (كائن arg0) |
| الدرجة النهائية<?> | الحصول على كلاس () |
| كثافة العمليات | رمز التجزئة () |
| الفراغ النهائي | إعلام () |
| الفراغ النهائي | إعلام الكل () |
| خيط | إلى سلسلة () |
| الفراغ النهائي | انتظر (طويل arg0، int arg1) |
| الفراغ النهائي | انتظر (طويل arg0) |
| الفراغ النهائي | انتظر () |
| بيئة التنفيذ المجردة | بيئى () قم بإرجاع بيئة التنفيذ التي تم إنشاء هذه العملية فيها. |
| عملية مجردة |
الثوابت
السلسلة النهائية الثابتة العامة OP_NAME
اسم هذه العملية كما هو معروف بواسطة محرك TensorFlow الأساسي
القيمة الثابتة: "CollectivePermute"
الأساليب العامة
الإخراج العام <T> كإخراج ()
إرجاع المقبض الرمزي للموتر.
المدخلات إلى عمليات TensorFlow هي مخرجات عملية TensorFlow أخرى. يتم استخدام هذه الطريقة للحصول على مقبض رمزي يمثل حساب الإدخال.
إنشاء CollectivePermute ثابت عام <T> ( نطاق النطاق ، وإدخال المعامل <T>، والمعامل < TInt32 > sourceTargetPairs)
طريقة المصنع لإنشاء فئة تغلف عملية CollectivePermute جديدة.
حدود
| نِطَاق | النطاق الحالي |
|---|---|
| مدخل | المدخلات المحلية المراد تبديلها. حاليًا يدعم فقط float وbfloat16. |
| sourceTargetPairs | موتر ذو شكل [num_pairs, 2]. |
عائدات
- مثيل جديد من CollectivePermute