تانسور ورودی را در تمام ابعاد تقسیم می کند.
عملیاتی که تانسور ورودی را بر اساس ویژگی 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
لایه های XlaSplitND.Options استاتیک عمومی (List<Long> paddings)
مولفه های
بالشتک ها | فهرست اختیاری لایههای سمت راست در هر بعد تانسور ورودی برای اعمال قبل از تقسیم. از این می توان برای تقسیم یک بعد یک بعد استفاده کرد. |
---|