XlaSplitND

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

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

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

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

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

[[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>، طولانی 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)

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