CollectivePermute

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

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

هر نمونه ورودی خود را ارائه می کند.

برای مثال، فرض کنید 4 نمونه TPU وجود دارد: «[A، B، C، D]». ارسال source_target_pairs=`[[0,1],[1,2],[2,3],[3,0]]` خروجی ها را دریافت می کند: `[D, A, B, C]`.

ثابت ها

رشته OP_NAME نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

روش های عمومی

خروجی <T>
asOutput ()
دسته نمادین تانسور را برمی گرداند.
static <T TType > CollectivePermute <T> را گسترش می دهد
ایجاد ( دامنه دامنه ، ورودی عملوند <T>، عملوند < TINT32 > SourceTargetPairs)
روش کارخانه برای ایجاد کلاسی که یک عملیات CollectivePermute جدید را بسته بندی می کند.
خروجی <T>
خروجی ()
ورودی جایگشت شده

روش های ارثی

ثابت ها

رشته نهایی ثابت عمومی OP_NAME

نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود

مقدار ثابت: "CollectivePermute"

روش های عمومی

خروجی عمومی <T> asOutput ()

دسته نمادین تانسور را برمی گرداند.

ورودی های عملیات TensorFlow خروجی های عملیات تنسورفلو دیگر هستند. این روش برای به دست آوردن یک دسته نمادین که نشان دهنده محاسبه ورودی است استفاده می شود.

عمومی استاتیک CollectivePermute <T> ایجاد ( دامنه دامنه ، ورودی عملوند <T>، عملوند < TINT32 > sourceTargetPairs)

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

مولفه های
محدوده محدوده فعلی
ورودی ورودی محلی که باید جایگزین شود. در حال حاضر فقط float و bfloat16 را پشتیبانی می کند.
sourceTargetPairs یک تانسور با شکل [num_pairs, 2].
برمی گرداند
  • یک نمونه جدید از CollectivePermute

خروجی عمومی <T> خروجی ()

ورودی جایگشت شده