XlaSplitND

کلاس نهایی عمومی XlaSplitND

تانسور ورودی را در تمام ابعاد تقسیم می کند.

عملیاتی که تانسور ورودی را بر اساس ویژگی num_splits داده شده برش می دهد، به صورت اختیاری برش می دهد و برش ها را برمی گرداند. برش ها به ترتیب ردیف اصلی برگردانده می شوند.

این عملیات ممکن است از طریق پل TPU ایجاد شود.

برای مثال، با تانسور «ورودی»:

[[0, 1, 2],
  [3, 4, 5],
  [6, 7, 8]]
"تعداد_شکاف":
[2, 2]
و "بالشتک":
[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>، طولانی N، فهرست <طولانی numSplits، گزینه‌ها... گزینه‌ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات XlaSplitND جدید را بسته بندی می کند.
Iterator< عملوند <T>>
تکرار کننده ()
لیست< خروجی <T>>
خروجی ها ()
استاتیک XlaSplitND.Options
بالشتک ها (List<Long> paddings)

روش های ارثی

روش های عمومی

ایستا عمومی XlaSplitND <T> ایجاد (حوزه دامنه ، ورودی عملوند <T>، Long N، لیست <طولانی> numSplits، گزینه‌ها... گزینه‌ها)

روش کارخانه برای ایجاد کلاسی که یک عملیات XlaSplitND جدید را بسته بندی می کند.

پارامترها
دامنه محدوده فعلی
ورودی تانسور ورودی برای تقسیم در تمام ابعاد. } out_arg { name: "outputs" توضیحات: <
numSplits تعداد روش های تقسیم در هر بعد. ابعاد شکل باید به طور مساوی قابل تقسیم باشد.
گزینه ها مقادیر ویژگی های اختیاری را حمل می کند
برمی گرداند
  • یک نمونه جدید از XlaSplitND

عمومی Iterator< عملوند <T>> تکرار کننده ()

فهرست عمومی< خروجی <T>> خروجی ها ()

لایه های XlaSplitND.Options استاتیک عمومی (List<Long> paddings)

پارامترها
بالشتک ها فهرست اختیاری لایه‌های سمت راست در هر بعد تانسور ورودی برای اعمال قبل از تقسیم. از این می توان برای تقسیم یک بعد یک بعد استفاده کرد.