CollectivePermute

الفئة النهائية العامة 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>، المعامل <TInt32> sourceTargetPairs)
طريقة المصنع لإنشاء فئة تغلف عملية CollectivePermute جديدة.
الإخراج <T>
انتاج ()
المدخلات المعدلة.

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

الثوابت

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

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

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

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

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

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

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

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

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

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

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

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