ApproxTopK

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

مقادیر 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' است.