الطبقة النهائية العامة ApproxTopK
تُرجع قيم min/max k ومؤشراتها لمعامل الإدخال بطريقة تقريبية.
راجع https://arxiv.org/abs/2206.14286 للحصول على تفاصيل الخوارزمية. تم تحسين هذه العملية فقط على مادة TPU حاليًا.
فئات متداخلة
| فصل | ApproxTopK.Options | السمات الاختيارية لـ ApproxTopK | |
الأساليب العامة
| ApproxTopK.Options ثابت | مجاميع توتوبك (إجمالي منطقي توتوبك) |
| ثابت <T يمتد الرقم> ApproxTopK <T> | |
| الإخراج <عدد صحيح> | المؤشرات () مؤشرات "القيم" على طول "بعد_التخفيض" لمعامل "الإدخال". |
| ApproxTopK.Options ثابت | isMaxK (المنطقية isMaxK) |
| ApproxTopK.Options ثابت | RecallTarget (تعويم RecallTarget) |
| ApproxTopK.Options ثابت | البعد التخفيض (البعد التخفيض الطويل) |
| ApproxTopK.Options ثابت | التخفيضInputSizeOverride (التخفيض الطويلInputSizeOverride) |
| الإخراج <T> | قيم () قيم الحد الأدنى/الأقصى k على طول "بعد_التخفيض" لمعامل "الإدخال". |
الطرق الموروثة
الأساليب العامة
ثابت عام ApproxTopK.Options مجاميع توتوبك (منطقي مجاميع توتوبك)
حدود
| complexToTopk | عندما يكون صحيحًا، يتم تجميع النتائج التقريبية إلى top-k. عند الخطأ، يتم إرجاع النتائج التقريبية. عدد النتائج التقريبية التي تم تحديدها من خلال التنفيذ وأكبر يساوي `k` المحدد. |
|---|
إنشاء ApproxTopK <T> ثابت عام ( نطاق النطاق ، وإدخال المعامل <T>، وLong k، وخيارات... خيارات)
طريقة المصنع لإنشاء فئة تلتف حول عملية ApproxTopK جديدة.
حدود
| نِطَاق | النطاق الحالي |
|---|---|
| مدخل | مصفوفة للبحث. يجب أن يكون على الأقل 1-D من النوع العائم |
| ك | يحدد عدد الحد الأدنى/الحد الأقصى-ك. |
| خيارات | يحمل قيم السمات الاختيارية |
عائدات
- مثيل جديد لـ ApproxTopK
ApproxTopK.Options العام الثابت هوMaxK (المنطقي isMaxK)
حدود
| isMaxK | عندما يكون صحيحا، يحسب max-k؛ وإلا فإنه يحسب min-k. |
|---|
الاستدعاء العام الثابت ApproxTopK.Options (الاستدعاء العائم)
حدود
| RecallTarget | أذكر الهدف للتقريب. النطاق في (0,1) |
|---|
ApproxTopK.Options التخفيض العام الثابت (البعد الطويل)
حدود
| البعد التخفيض | البعد الصحيح الذي سيتم البحث من خلاله. الافتراضي: -1. |
|---|
ثابت عام ApproxTopK.Options تخفيضInputSizeOverride (تخفيض طويلInputSizeOverride)
حدود
| ReductionInputSizeOverride | عند التعيين على قيمة موجبة، فإنه يتجاوز الحجم المحدد بواسطة `input[reduction_dim]` لتقييم الاستدعاء. يكون هذا الخيار مفيدًا عندما يكون "الإدخال" المحدد مجرد مجموعة فرعية من الحساب الإجمالي في SPMD أو خطوط الأنابيب الموزعة، حيث لا يمكن تأجيل حجم الإدخال الحقيقي بواسطة شكل "الإدخال". |
|---|
قيم الإخراج العامة <T> ()
قيم الحد الأدنى/الأقصى k على طول "بعد_التخفيض" لمعامل "الإدخال". البعد هو نفس معامل "الإدخال" باستثناء "بعد_التخفيض": عندما يكون "aggregate_to_topk" صحيحًا، يكون بُعد التخفيض هو "k"؛ وإلا فإنه يساوي أكبر `k` حيث يكون الحجم محددًا بالتنفيذ.