सार्वजनिक अंतिम वर्ग
न्यूनतम/अधिकतम k मान और इनपुट ऑपरेंड के उनके सूचकांकों को अनुमानित तरीके से लौटाता है।
एल्गोरिथम विवरण के लिए https://arxiv.org/abs/2206.14286 देखें। यह ऑप वर्तमान में केवल टीपीयू पर अनुकूलित है।
नेस्टेड कक्षाएं
| कक्षा | लगभग TopK.Options | ApproxTopK के लिए वैकल्पिक विशेषताएँ | |
सार्वजनिक तरीके
| स्थिर लगभग TopK.Options | एग्रीगेट टोटॉप (बूलियन एग्रीगेट टोटॉप) |
| स्थिर <टी संख्या बढ़ाता है> लगभग टॉपके <टी> | |
| आउटपुट <पूर्णांक> | सूचकांक () `इनपुट` ऑपरेंड के `reduction_dimension` के साथ `मान` के सूचकांक। |
| स्थिर लगभग TopK.Options | isMaxK (बूलियन isMaxK) |
| स्थिर लगभग TopK.Options | रिकॉल टारगेट (फ्लोट रिकॉल टारगेट) |
| स्थिर लगभग TopK.Options | कमीआयाम (लंबी कमीआयाम) |
| स्थिर लगभग TopK.Options | कमी इनपुट आकार ओवरराइड (लंबी कमी इनपुट आकार ओवरराइड) |
| आउटपुट <टी> | मान () `इनपुट` संकार्य के `reduction_dimension` के साथ न्यूनतम/अधिकतम k मान। |
विरासत में मिली विधियां
सार्वजनिक तरीके
सार्वजनिक स्थैतिक ApproxTopK.Options समुच्चय ToTopk (बूलियन समुच्चयToTopk)
मापदंडों
| समुच्चयतोटॉपक | जब सही होता है, तो अनुमानित परिणामों को टॉप-के में एकत्रित करता है। गलत होने पर, अनुमानित परिणाम लौटाता है। अनुमानित परिणामों की संख्या कार्यान्वयन परिभाषित है और निर्दिष्ट `के` के बराबर है। |
|---|
सार्वजनिक स्थैतिक ApproxTopK <T> बनाएँ ( स्कोप स्कोप, ऑपरेंड <T> इनपुट, लॉन्ग k, विकल्प ... विकल्प)
एक नया ApproxTopK ऑपरेशन रैपिंग क्लास बनाने के लिए फ़ैक्टरी विधि।
मापदंडों
| दायरा | वर्तमान दायरा |
|---|---|
| इनपुट | खोजने के लिए सरणी। फ़्लोटिंग प्रकार का कम से कम 1-डी होना चाहिए |
| क | न्यूनतम/अधिकतम-के की संख्या निर्दिष्ट करता है। |
| विकल्प | वैकल्पिक विशेषता मान रखता है |
रिटर्न
- ApproxTopK का एक नया उदाहरण
सार्वजनिक आउटपुट <पूर्णांक> सूचकांक ()
`इनपुट` ऑपरेंड के `reduction_dimension` के साथ `मान` के सूचकांक।
सार्वजनिक स्थैतिक ApproxTopK.Options isMaxK (बूलियन isMaxK)
मापदंडों
| isMaxK | जब सत्य, अधिकतम-के की गणना करता है; अन्यथा मिन-के की गणना करता है। |
|---|
सार्वजनिक स्थैतिक ApproxTopK.Options याद लक्ष्य (फ्लोट रिकॉल लक्ष्य)
मापदंडों
| याद लक्ष्य | सन्निकटन के लिए लक्ष्य को याद करें। में रेंज (0,1] |
|---|
सार्वजनिक स्थैतिक ApproxTopK.Options कमीआयाम ( लंबी कमीआयाम)
मापदंडों
| कमी आयाम | पूर्णांक आयाम जिसके साथ खोजना है। डिफ़ॉल्ट: -1। |
|---|
सार्वजनिक स्थैतिक ApproxTopK.Options कमी InputSizeOverride (लंबी कमीInputSizeOverride)
मापदंडों
| कमीइनपुटसाइजओवरराइड | जब एक सकारात्मक मान पर सेट किया जाता है, तो यह रिकॉल के मूल्यांकन के लिए `इनपुट [reduction_dim]` द्वारा निर्धारित आकार को ओवरराइड करता है। यह विकल्प तब उपयोगी होता है जब दिया गया `इनपुट` एसपीएमडी या वितरित पाइपलाइनों में समग्र गणना का केवल एक सबसेट होता है, जहां सही इनपुट आकार को `इनपुट` आकार से स्थगित नहीं किया जा सकता है। |
|---|
सार्वजनिक आउटपुट <T> मान ()
`इनपुट` संकार्य के `reduction_dimension` के साथ न्यूनतम/अधिकतम k मान। आयाम `इनपुट` ऑपरेंड के समान हैं, केवल `reduction_dimension` को छोड़कर: जब `aggregate_to_topk` सत्य है, तो कमी आयाम `k` है; अन्यथा, यह `के` के बराबर है जहां आकार कार्यान्वयन-परिभाषित है।