अनुमानित तरीके से इनपुट ऑपरेंड के न्यूनतम/अधिकतम k मान और उनके सूचकांक लौटाता है।
एल्गोरिथम विवरण के लिए https://arxiv.org/abs/2206.14286 देखें। यह ऑप अभी केवल टीपीयू पर अनुकूलित है।
नेस्टेड क्लासेस
कक्षा | ApproxTopK.विकल्प | ApproxTopK के लिए वैकल्पिक विशेषताएँ |
सार्वजनिक तरीके
स्थिर ApproxTopK.Options | एग्रीगेटटूटॉपक (बूलियन एग्रीगेटटूटॉपक) |
स्थिर <T संख्या बढ़ाता है> लगभग शीर्ष <T> | |
आउटपुट <पूर्णांक> | सूचकांक () `इनपुट` ऑपरेंड के `रिडक्शन_डायमेंशन` के साथ `मानों` के सूचकांक। |
स्थिर 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` के बराबर है जहां आकार कार्यान्वयन-परिभाषित है।