مقادیر min/max k و شاخص های آنها را از عملوند ورودی به روشی تقریبی برمی گرداند.
برای جزئیات الگوریتم به https://arxiv.org/abs/2206.14286 مراجعه کنید. این عملیات در حال حاضر فقط در TPU بهینه شده است.
کلاس های تو در تو
کلاس | ApproxTopK.Options | ویژگی های اختیاری برای ApproxTopK |
روش های عمومی
استاتیک ApproxTopK.Options | aggregateToTopk (AggregateToTopk Boolean) |
استاتیک <T گسترش شماره> ApproxTopK <T> | ایجاد (حوزه دامنه ، ورودی عملوند <T>، Long k، گزینهها... گزینهها) روش کارخانه برای ایجاد کلاسی که یک عملیات ApproxTopK جدید را بسته بندی می کند. |
خروجی <عدد صحیح> | شاخص ها () شاخصهای «مقادیر» در امتداد «بعد_کاهشی» عملوند «ورودی». |
استاتیک ApproxTopK.Options | isMaxK (بولی isMaxK) |
استاتیک ApproxTopK.Options | recallTarget (Float recallTarget) |
استاتیک ApproxTopK.Options | ابعاد کاهشی (بعد کاهش طولانی) |
استاتیک ApproxTopK.Options | reduceInputSizeOverride (مدت کاهشInputSizeOverride) |
خروجی <T> | ارزش های () مقدار min/max k در امتداد «reduction_dimension» عملوند «ورودی» است. |
روش های ارثی
روش های عمومی
استاتیک عمومی ApproxTopK.Options aggregateToTopk (AggregateToTopk بولی)
مولفه های
aggregateToTopk | وقتی درست است، مجموع ها نتایج را به top-k تقریبی می کنند. وقتی غلط باشد، نتایج تقریبی را برمیگرداند. تعداد نتایج تقریبی پیاده سازی شده است و بیشتر برابر با «k» مشخص شده است. |
---|
استاتیک عمومی ApproxTopK <T> ایجاد (حوزه دامنه ، ورودی عملوند <T>، Long k، گزینهها... گزینهها)
روش کارخانه برای ایجاد کلاسی که یک عملیات ApproxTopK جدید را بسته بندی می کند.
مولفه های
محدوده | محدوده فعلی |
---|---|
ورودی | آرایه برای جستجو. باید حداقل 1-D از نوع شناور باشد |
ک | تعداد min/max-k را مشخص می کند. |
گزینه ها | مقادیر ویژگی های اختیاری را حمل می کند |
برمی گرداند
- یک نمونه جدید از ApproxTopK
عمومی استاتیک ApproxTopK.Options isMaxK (Boolean isMaxK)
مولفه های
isMaxK | وقتی درست است، max-k را محاسبه می کند. در غیر این صورت min-k را محاسبه می کند. |
---|
عمومی استاتیک ApproxTopK.Options recallTarget (Float recallTarget)
مولفه های
recallTarget | هدف را برای تقریب به یاد بیاورید. محدوده در (0,1] |
---|
عمومی استاتیک ApproxTopK.Options reduceDimension (Long reduceDimension)
مولفه های
کاهش ابعاد | بعد عدد صحیح که در امتداد آن جستجو می شود. پیش فرض: -1. |
---|
عمومی ایستا ApproxTopK.Options reduceInputSizeOverride (مدت کاهشInputSizeOverride)
مولفه های
reduceInputSizeOverride | وقتی روی یک مقدار مثبت تنظیم میشود، اندازه تعیینشده توسط «input[reduction_dim]» برای ارزیابی فراخوان لغو میشود. این گزینه زمانی مفید است که «ورودی» داده شده تنها زیرمجموعه ای از محاسبات کلی در SPMD یا خطوط لوله توزیع شده باشد، جایی که اندازه ورودی واقعی را نمی توان با شکل «ورودی» به تعویق انداخت. |
---|
مقادیر <T> خروجی عمومی ()
مقدار min/max k در امتداد «reduction_dimension» عملوند «ورودی» است. بعد همان عملوند "ورودی" است به جز "reduction_dimension": وقتی "Aggregate_to_topk" درست است، بعد کاهشی "k" است. در غیر این صورت، در جایی که اندازه آن توسط پیاده سازی تعریف شده است، برابر با 'k' است.