tensorflow:: ops:: TopK

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

सारांश

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

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

values.shape = indices.shape = input.shape[:-1] + [k]

यदि दो तत्व बराबर हैं, तो कम सूचकांक वाला तत्व पहले आता है।

तर्क:

• स्कोप: एक स्कोप ऑब्जेक्ट
• इनपुट: 1-डी या उससे अधिक, जिसमें अंतिम आयाम कम से कम k हो।
• k: 0-D. अंतिम आयाम के साथ (मैट्रिक्स के लिए प्रत्येक पंक्ति के साथ) खोजे जाने वाले शीर्ष तत्वों की संख्या।

वैकल्पिक विशेषताएँ (देखें Attrs ):

• सॉर्टेड: यदि सत्य है, तो परिणामी k तत्वों को मानों के आधार पर अवरोही क्रम में सॉर्ट किया जाएगा।

प्रतिफल:

• Output मान: प्रत्येक अंतिम आयामी स्लाइस के साथ k सबसे बड़े तत्व।
• Output इंडेक्स: input के अंतिम आयाम के भीतर values के इंडेक्स।

सार्वजनिक विशेषताएँ

::tensorflow::Output indices

Operation operation

::tensorflow::Output values

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k
)

 TopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input k,
  const TopK::Attrs & attrs
)

Attrs Sorted(
  bool x
)