ApproxTopK

सार्वजनिक अंतिम कक्षा ApproxTopK

अनुमानित तरीके से इनपुट ऑपरेंड के न्यूनतम/अधिकतम k मान और उनके सूचकांक लौटाता है।

एल्गोरिथम विवरण के लिए https://arxiv.org/abs/2206.14286 देखें। यह ऑप अभी केवल टीपीयू पर अनुकूलित है।

नेस्टेड क्लासेस

कक्षा ApproxTopK.विकल्प ApproxTopK के लिए वैकल्पिक विशेषताएँ

सार्वजनिक तरीके

स्थिर ApproxTopK.Options
एग्रीगेटटूटॉपक (बूलियन एग्रीगेटटूटॉपक)
स्थिर <T संख्या बढ़ाता है> लगभग शीर्ष <T>
बनाएं ( स्कोप स्कोप, ऑपरेंड <टी> इनपुट, लॉन्ग के, विकल्प... विकल्प)
एक नया ApproxTopK ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।
आउटपुट <पूर्णांक>
सूचकांक ()
`इनपुट` ऑपरेंड के `रिडक्शन_डायमेंशन` के साथ `मानों` के सूचकांक।
स्थिर ApproxTopK.Options
isMaxK (बूलियन isMaxK)
स्थिर ApproxTopK.Options
रिकॉलटार्गेट (फ्लोट रिकॉलटार्गेट)
स्थिर ApproxTopK.Options
कमी आयाम (लंबी कमी आयाम)
स्थिर ApproxTopK.Options
रिडक्शनइनपुटसाइजओवरराइड (लंबा रिडक्शनइनपुटसाइजओवरराइड)
आउटपुट <T>
मान ()
`इनपुट` ऑपरेंड के `रिडक्शन_डायमेंशन` के साथ न्यूनतम/अधिकतम k मान।

विरासत में मिली विधियाँ

सार्वजनिक तरीके

सार्वजनिक स्थैतिक ApproxTopK.Options समुच्चयToTopk (बूलियन समुच्चयToTopk)

पैरामीटर
समुच्चयToTopk सही होने पर, अनुमानित परिणामों को टॉप-के पर एकत्रित करता है। गलत होने पर, अनुमानित परिणाम लौटाता है। अनुमानित परिणामों की संख्या कार्यान्वयन परिभाषित है और निर्दिष्ट `k` के बराबर अधिक है।

सार्वजनिक स्थैतिक ApproxTopK <T> बनाएं ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, लॉन्ग k, विकल्प... विकल्प)

एक नया ApproxTopK ऑपरेशन लपेटकर एक क्लास बनाने की फ़ैक्टरी विधि।

पैरामीटर
दायरा वर्तमान दायरा
इनपुट खोजने के लिए सरणी. फ्लोटिंग प्रकार का कम से कम 1-डी होना चाहिए
न्यूनतम/अधिकतम-k की संख्या निर्दिष्ट करता है।
विकल्प वैकल्पिक गुण मान रखता है
रिटर्न
  • ApproxTopK का एक नया उदाहरण

सार्वजनिक आउटपुट <पूर्णांक> सूचकांक ()

`इनपुट` ऑपरेंड के `रिडक्शन_डायमेंशन` के साथ `मानों` के सूचकांक।

सार्वजनिक स्थैतिक ApproxTopK.Options isMaxK (बूलियन isMaxK)

पैरामीटर
मैक्सके है सत्य होने पर, max-k की गणना करता है; अन्यथा min-k की गणना करता है।

सार्वजनिक स्थैतिक ApproxTopK.ऑप्शंस रिकॉलटार्गेट (फ्लोट रिकॉलटार्गेट)

पैरामीटर
पुनः लक्ष्य सन्निकटन के लिए लक्ष्य को याद करें। रेंज इन (0,1]

सार्वजनिक स्थैतिक ApproxTopK.विकल्प कमी आयाम (लंबी कमी आयाम)

पैरामीटर
कमी आयाम पूर्णांक आयाम जिसके साथ खोजना है. डिफ़ॉल्ट: -1.

सार्वजनिक स्थैतिक ApproxTopK.विकल्प कमी इनपुट आकार ओवरराइड (लंबी कमी इनपुट आकार ओवरराइड)

पैरामीटर
कमी इनपुट आकार ओवरराइड जब इसे सकारात्मक मान पर सेट किया जाता है, तो यह रिकॉल के मूल्यांकन के लिए `input[reduction_dim]` द्वारा निर्धारित आकार को ओवरराइड कर देता है। यह विकल्प तब उपयोगी होता है जब दिया गया `इनपुट` एसपीएमडी या वितरित पाइपलाइनों में समग्र गणना का केवल एक सबसेट होता है, जहां वास्तविक इनपुट आकार को `इनपुट` आकार द्वारा स्थगित नहीं किया जा सकता है।

सार्वजनिक आउटपुट <T> मान ()

`इनपुट` ऑपरेंड के `रिडक्शन_डायमेंशन` के साथ न्यूनतम/अधिकतम k मान। `reduction_dimension` को छोड़कर आयाम `इनपुट` ऑपरेंड के समान है: जब `aggregate_to_topk` सत्य है, तो कमी का आयाम `k` है; अन्यथा, यह `k` के बराबर है जहां आकार कार्यान्वयन-परिभाषित है।