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

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

#include <nn_ops.h>

कहते हैं कि क्या लक्ष्य शीर्ष K भविष्यवाणियों में हैं।

सारांश

यह एक आउटपुट batch_size bool सरणी, एक प्रवेश out[i] है true है, तो लक्ष्य वर्ग के लिए भविष्यवाणी शीर्ष के बीच में है k उदाहरण के लिए सभी भविष्यवाणियों के बीच भविष्यवाणियों i । ध्यान दें कि InTopK का व्यवहार संबंधों को संभालने में TopK op से भिन्न होता है; यदि कई वर्गों का एक ही पूर्वानुमान मूल्य है और शीर्ष- k सीमा को काटते हैं, तो उन सभी वर्गों को शीर्ष k में माना जाता है।

अधिक औपचारिक रूप से, चलो

\(predictions_i\) हो उदाहरण के लिए सभी वर्गों के लिए भविष्यवाणियों i , उदाहरण के लिए लक्ष्य वर्ग \(targets_i\) i , \(out_i\) उत्पादन उदाहरण के लिए i ,

$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$

तर्क:

  • गुंजाइश: एक स्कोप ऑब्जेक्ट
  • भविष्यवाणियां: एक batch_size एक्स classes टेंसर।
  • लक्ष्य: कक्षा आईडी का एक batch_size वेक्टर।
  • k: कंप्यूटिंग परिशुद्धता के लिए देखने के लिए शीर्ष तत्वों की संख्या।

रिटर्न:

  • Output : एक bool Tensor रूप में k पर कम्प्यूटेड प्रिसिजन।

कंस्ट्रक्टर और डिस्ट्रक्टर्स

InTopK (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, int64 k)

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

operation
precision

सार्वजनिक कार्य

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

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

ऑपरेशन

Operation operation

शुद्धता

::tensorflow::Output precision

सार्वजनिक कार्य

InTopK

 InTopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  int64 k
)
है

नोड

::tensorflow::Node * node() const 

ऑपरेटर :: टेंसरफ़्लो :: इनपुट

 operator::tensorflow::Input() const 
है

ऑपरेटर :: टेंसरफ़्लो :: आउटपुट

 operator::tensorflow::Output() const