Google I/O में ट्यूनिंग के लिए धन्यवाद। मांग पर सभी सत्र देखें मांग पर देखें

टेंसरफ़्लो :: ऑप्स :: टॉपके

#include <nn_ops.h>

पाता मूल्यों और के सूचकांकों k पिछले आयाम के लिए सबसे बड़ा तत्व।

सारांश

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

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

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

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

तर्क:

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

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

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

रिटर्न:

  • Output मान: k प्रत्येक पिछले आयामी टुकड़ा के साथ सबसे बड़ा तत्व।
  • Output सूचकांक: input के अंतिम आयाम के भीतर 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)

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

indices
operation
values

सार्वजनिक स्थैतिक कार्य

Sorted (bool x)

संरचनाएं

टेंसोफ़्लो :: ऑप्स :: टॉपके :: एट्र्स

वैकल्पिक विशेषता TopK के लिए बसती है

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

सूचकांक

::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
)