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