TPUPartitionedOutputV2

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

عملیاتی که یک تانسور را از هم مولتیپلکس می کند تا توسط XLA به لیستی از پارتیشن بندی شده تقسیم شود

خروجی های خارج از محاسبات XLA از اشتراک گذاری ND پشتیبانی می کند.

روش های عمومی

استاتیک <T> TPUPpartitionedOutputV2 <T>
ایجاد (حوزه دامنه ، ورودی های Operand <T>، NumSplits طولانی، List<Long> partitionDims)
روش کارخانه برای ایجاد کلاسی که عملیات جدید TPUPpartitionedOutputV2 را بسته بندی می کند.
Iterator< عملوند <T>>
لیست< خروجی <T>>
خروجی ()
لیستی از خروجی های پارتیشن بندی شده که شکل یکسانی دارند.

روش های ارثی

روش های عمومی

عمومی ایستا TPUPpartitionedOutputV2 <T> ایجاد ( دامنه دامنه ، ورودی های عملوند <T>، NumSplits طولانی، List<Long> partitionDims)

روش کارخانه برای ایجاد کلاسی که عملیات جدید TPUPpartitionedOutputV2 را بسته بندی می کند.

مولفه های
محدوده محدوده فعلی
ورودی ها تانسوری که نمایانگر شکل کامل تانسورهای تقسیم شده است.
partitionDims فهرستی از اعداد صحیح که نحوه تقسیم بندی هر بعد را توضیح می دهد. خالی بودن نشان می دهد که ورودی ها تکرار می شوند.
برمی گرداند
  • یک نمونه جدید از TPUPpartitionedOutputV2

عمومی Iterator< عملوند <T>> تکرار کننده ()

فهرست عمومی< خروجی <T>> خروجی ()

لیستی از خروجی های پارتیشن بندی شده که شکل یکسانی دارند.