کلاس نهایی عمومی 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> را گسترش می دهد | |
| خروجی < TINT32 > | شاخص ها () شاخص‌های «ارزش‌ها» در آخرین بعد «ورودی». |
| استاتیک TopK.Options | مرتب شده (مرتب شده به زبان بولی) |
| خروجی <T> | ارزش های () "k" بزرگترین عناصر در امتداد هر برش بعدی آخرین. |
روش های ارثی
ثابت ها
رشته نهایی ثابت عمومی OP_NAME
نام این عملیات، همانطور که توسط موتور هسته TensorFlow شناخته می شود
مقدار ثابت: "TopKV2"
روش های عمومی
عمومی استاتیک TopK <T> ایجاد ( دامنه دامنه ، ورودی عملوند <T>، عملوند < TINT32 > k، گزینه‌ها... گزینه‌ها)
روش کارخانه برای ایجاد کلاسی که یک عملیات TopK جدید را بسته بندی می کند.
مولفه های
| محدوده | محدوده فعلی |
|---|---|
| ورودی | 1-D یا بالاتر با آخرین بعد حداقل "k". |
| ک | 0-D. تعداد عناصر بالایی که باید در امتداد آخرین بعد (در امتداد هر ردیف برای ماتریس ها) جستجو شوند. |
| گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از TopK
عمومی استاتیک TopK. Options مرتب شده (مرتب شده بولین)
مولفه های
| مرتب شده است | اگر درست باشد، عناصر «k» حاصل بر اساس مقادیر به ترتیب نزولی مرتب خواهند شد. |
|---|