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

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

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

الطرق العامة

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

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

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

ساكنة العام CollectivePermute <T> إنشاء ( نطاق نطاق، المعامل <T> المدخلات، و المعامل <صحيح> sourceTargetPairs)

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

العوامل
مجال النطاق الحالي
إدخال المدخلات المحلية المراد تبديلها. يدعم حاليًا كل من float و bfloat16.
sourceTargetPairs موتر ذو شكل [عدد_أزواج ، 2].
عائدات
  • نسخة جديدة من CollectivePermute

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

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