TopK

अंतिम आयाम के लिए `k` सबसे बड़े तत्वों के मान और सूचकांक ढूँढता है।

यदि इनपुट एक वेक्टर (रैंक-1) है, तो वेक्टर में `k` सबसे बड़ी प्रविष्टियाँ ढूँढता है और उनके मूल्यों और सूचकांकों को वेक्टर के रूप में आउटपुट करता है। इस प्रकार `values[j]` `input` में `j`-th सबसे बड़ी प्रविष्टि है, और इसका सूचकांक `indices[j]` है।

मैट्रिक्स (उच्च रैंक इनपुट के संबंध में) के लिए, प्रत्येक पंक्ति में शीर्ष `k` प्रविष्टियों की गणना करता है (अंतिम आयाम के साथ संबंधित वेक्टर)। इस प्रकार,

मान.आकार = सूचकांक.आकार = इनपुट.आकार[:-1] + [के]

यदि दो तत्व समान हैं, तो निचला-सूचकांक तत्व पहले दिखाई देता है।