تانسور ورودی متغیر منبع را در همه ابعاد تقسیم می کند.
عملیاتی که تانسور ورودی متغیر منبع را بر اساس ویژگی 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]]
کلاس های تو در تو
کلاس | ReadVariableXlaSplitND.Options | ویژگی های اختیاری برای ReadVariableXlaSplitND |
روش های عمومی
استاتیک <T> ReadVariableXlaSplitND <T> | ایجاد (حوزه دامنه ، منبع عملوند <?>، Class<T> T، طولانی N، فهرست <طولانی numSplits، گزینهها... گزینهها) روش کارخانه برای ایجاد کلاسی که یک عملیات ReadVariableXlaSplitND جدید را بسته بندی می کند. |
Iterator< عملوند <T>> | تکرار کننده () |
فهرست< خروجی <T>> | خروجی ها () |
استاتیک ReadVariableXlaSplitND.Options | بالشتک ها (List<Long> paddings) |
روش های ارثی
روش های عمومی
استاتیک عمومی ReadVariableXlaSplitND <T> ایجاد ( محدوده دامنه ، منبع عملوند <?>، Class<T> T، طولانی N، فهرست <طولانی> numSplits، گزینهها... گزینهها)
روش کارخانه برای ایجاد کلاسی که یک عملیات ReadVariableXlaSplitND جدید را بسته بندی می کند.
پارامترها
دامنه | محدوده فعلی |
---|---|
منبع | متغیر منبع تانسور ورودی برای تقسیم در تمام ابعاد. } out_arg { name: "outputs" توضیحات: < |
numSplits | تعداد روش های تقسیم در هر بعد. ابعاد شکل باید به طور مساوی قابل تقسیم باشد. |
گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از ReadVariableXlaSplitND
استاتیک عمومی ReadVariableXlaSplitND.Options paddings (List<Long> paddings)
پارامترها
بالشتک ها | فهرست اختیاری لایههای سمت راست در هر بعد تانسور ورودی برای اعمال قبل از تقسیم. از این می توان برای تقسیم یک بعد یک بعد استفاده کرد. |
---|