CollectivePermute

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

عملية لتبديل الموترات عبر مثيلات TPU المكررة.

توفر كل حالة مدخلاتها الخاصة.

على سبيل المثال، لنفترض أن هناك 4 مثيلات لـ TPU: `[A, B, C, D]`. تمرير source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` يحصل على المخرجات: `[D, A, B, C]`.

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

الإخراج <T>
كإخراج ()
إرجاع المقبض الرمزي للموتر.
ثابت <T> CollectivePermute <T>
إنشاء (نطاق النطاق ، إدخال المعامل <T>، المعامل <Integer> sourceTargetPairs)
طريقة المصنع لإنشاء فئة تغلف عملية CollectivePermute جديدة.
الإخراج <T>
انتاج ()
المدخلات المعدلة.

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

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

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

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

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

إنشاء CollectivePermute ثابت عام <T> ( نطاق النطاق ، وإدخال المعامل <T>، والمعامل <Integer> sourceTargetPairs)

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

حدود
نِطَاق النطاق الحالي
مدخل المدخلات المحلية المراد تبديلها. حاليًا يدعم فقط float وbfloat16.
sourceTargetPairs موتر ذو شكل [num_pairs, 2].
عائدات
  • مثيل جديد من CollectivePermute

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

المدخلات المعدلة.