TopK

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

مقادیر و شاخص‌های «k» بزرگترین عناصر را برای آخرین بعد پیدا می‌کند.

اگر ورودی یک بردار باشد (رتبه-1)، بزرگترین ورودی های "k" را در بردار پیدا می کند و مقادیر و شاخص های آنها را به عنوان بردار خروجی می دهد. بنابراین «مقدار[j]» «j»-امین بزرگترین ورودی در «ورودی» است و شاخص آن «شاخص[j]» است.

برای ماتریس‌ها (به عبارت دیگر، ورودی رتبه بالاتر)، ورودی‌های «k» بالایی را در هر سطر محاسبه می‌کند (مثلاً بردار در امتداد آخرین بعد). بدین ترتیب،

values.shape = indices.shape = input.shape[:-1] + [k]

اگر دو عنصر برابر باشند، ابتدا عنصر با شاخص پایین ظاهر می شود.

کلاس های تو در تو

کلاس TopK.Options ویژگی های اختیاری برای TopK

ثابت ها

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

روش های عمومی

<T ثابت TNumber > TopK <T> را گسترش می دهد
ایجاد ( دامنه دامنه ، ورودی عملوند <T>، عملوند < TINT32 > k، گزینه‌ها... گزینه‌ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات TopK جدید را بسته بندی می کند.
خروجی < TINT32 >
شاخص ها ()
شاخص‌های «ارزش‌ها» در آخرین بعد «ورودی».
استاتیک TopK.Options
مرتب شده (مرتب شده به زبان بولی)
خروجی <T>
ارزش های ()
"k" بزرگترین عناصر در امتداد هر برش بعدی آخرین.

روش های ارثی

ثابت ها

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

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

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

روش های عمومی

عمومی استاتیک TopK <T> ایجاد ( دامنه دامنه ، ورودی عملوند <T>، عملوند < TINT32 > k، گزینه‌ها... گزینه‌ها)

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

مولفه های
محدوده محدوده فعلی
ورودی 1-D یا بالاتر با آخرین بعد حداقل "k".
ک 0-D. تعداد عناصر بالایی که باید در امتداد آخرین بعد (در امتداد هر ردیف برای ماتریس ها) جستجو شوند.
گزینه ها مقادیر ویژگی های اختیاری را حمل می کند
برمی گرداند
  • یک نمونه جدید از TopK

خروجی عمومی < TINT32 > شاخص ها ()

شاخص‌های «ارزش‌ها» در آخرین بعد «ورودی».

عمومی استاتیک TopK. Options مرتب شده (مرتب شده بولین)

مولفه های
مرتب شده است اگر درست باشد، عناصر «k» حاصل بر اساس مقادیر به ترتیب نزولی مرتب خواهند شد.

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

"k" بزرگترین عناصر در امتداد هر برش بعدی آخرین.