کلاس نهایی عمومی SplitV
یک تانسور را در امتداد یک بعد به تانسورهای «num_split» تقسیم میکند.
روش های عمومی
استاتیک <T، U گسترش شماره> SplitV <T> | |
Iterator< عملوند <T>> | اشاره گر () |
لیست< خروجی <T>> | خروجی () تانسورهایی که شکل آنها با "مقدار" مطابقت دارد، به جز در امتداد "محور"، جایی که اندازه آنها "size_splits[i]" است. |
روش های ارثی
روش های عمومی
ایجاد SplitV ایستا عمومی <T> ( scope scope، Operand <T> مقدار، Operand <U> sizeSplits، Operand <Integer> axis، Long numSplit)
روش کارخانه برای ایجاد کلاسی که عملیات SplitV جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
ارزش | تانسور برای شکافتن. |
sizeSplits | فهرستی حاوی اندازه های هر تانسور خروجی در امتداد بعد تقسیم. باید به بعد مقدار در امتداد split_dim جمع شود. می تواند حاوی یک -1 باشد که نشان می دهد بعد باید استنباط شود. |
محور | 0-D. بعد که در امتداد آن تقسیم شود. باید در محدوده "[-rank(value)، rank(value))" باشد. |
برمی گرداند
- یک نمونه جدید از SplitV
فهرست عمومی< خروجی <T>> خروجی ()
تانسورهایی که شکل آنها با "مقدار" مطابقت دارد، به جز در امتداد "محور"، جایی که اندازه آنها "size_splits[i]" است.