XlaSplitND

الطبقة النهائية العامة XlaSplitND

يقسم موتر الإدخال عبر جميع الأبعاد.

عملية تقوم بتقطيع موتر الإدخال استنادًا إلى سمة 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

مكرر عام< Operand <T>> مكرر ()

القائمة العامة< الإخراج <T>> النواتج ()

حشوات XlaSplitND.Options العامة الثابتة (حشوات القائمة <Long>)

حدود
الحشوات قائمة اختيارية من الحشوات الصحيحة لكل بُعد من موتر الإدخال لتطبيقها قبل التقسيم. يمكن استخدام هذا لجعل البعد قابلاً للقسمة بالتساوي.