يقسم موتر الإدخال عبر جميع الأبعاد.
عملية تقوم بتقطيع موتر الإدخال استنادًا إلى سمة num_splits المعطاة، وشرائح الوسادات بشكل اختياري، وإرجاع الشرائح. يتم إرجاع الشرائح بترتيب الصف الرئيسي.
قد يتم إنشاء هذه العملية عبر جسر TPU.
على سبيل المثال، مع الموتر `input`:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
`num_splits`: [2, 2]
و`paddings`: [1, 1]
فإن `المخرجات` المتوقعة هي: [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
فئات متداخلة
فصل | XlaSplitND.Options | السمات الاختيارية لـ XlaSplitND |
الأساليب العامة
ثابت <T> XlaSplitND <T> | إنشاء (نطاق النطاق ، إدخال المعامل <T>، Long N، List<Long> numSplits، Options... options) طريقة المصنع لإنشاء فئة تغلف عملية XlaSplitND جديدة. |
التكرار< المعامل <T>> | مكرر () |
القائمة< الإخراج <T>> | المخرجات () |
XlaSplitND.Options ثابت | الحشوات (حشوات القائمة<Long>) |
الطرق الموروثة
الأساليب العامة
إنشاء XlaSplitND <T> ثابت عام (نطاق النطاق ، وإدخال المعامل <T>، Long N، List<Long> numSplits، Options... options)
طريقة المصنع لإنشاء فئة تغلف عملية XlaSplitND جديدة.
حدود
نِطَاق | النطاق الحالي |
---|---|
مدخل | موتر الإدخال للتقسيم عبر جميع الأبعاد. } out_arg { الاسم: وصف "المخرجات": < |
numSplits | عدد طرق التقسيم لكل بُعد. يجب أن تكون أبعاد الشكل قابلة للقسمة بالتساوي. |
خيارات | يحمل قيم السمات الاختيارية |
عائدات
- مثيل جديد لـ XlaSplitND
حشوات XlaSplitND.Options العامة الثابتة (حشوات القائمة <Long>)
حدود
الحشوات | قائمة اختيارية من الحشوات الصحيحة لكل بُعد من موتر الإدخال لتطبيقها قبل التقسيم. يمكن استخدام هذا لجعل البعد قابلاً للقسمة بالتساوي. |
---|