کلاس نهایی عمومی TUPpartitionedOutputV2
عملیاتی که یک تانسور را از هم مولتیپلکس می کند تا توسط XLA به لیستی از پارتیشن بندی شده تقسیم شود
خروجی های خارج از محاسبات XLA از اشتراک گذاری ND پشتیبانی می کند.
روش های عمومی
استاتیک <T> TPUPpartitionedOutputV2 <T> | |
Iterator< عملوند <T>> | اشاره گر () |
لیست< خروجی <T>> | خروجی () لیستی از خروجی های پارتیشن بندی شده که شکل یکسانی دارند. |
روش های ارثی
روش های عمومی
عمومی ایستا TPUPpartitionedOutputV2 <T> ایجاد ( دامنه دامنه ، ورودی های عملوند <T>، NumSplits طولانی، List<Long> partitionDims)
روش کارخانه برای ایجاد کلاسی که عملیات جدید TPUPpartitionedOutputV2 را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
ورودی ها | تانسوری که نمایانگر شکل کامل تانسورهای تقسیم شده است. |
partitionDims | فهرستی از اعداد صحیح که نحوه تقسیم بندی هر بعد را توضیح می دهد. خالی بودن نشان می دهد که ورودی ها تکرار می شوند. |
برمی گرداند
- یک نمونه جدید از TPUPpartitionedOutputV2