टेंसरफ़्लो :: ऑप्स :: InTopKV2
#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: कंप्यूटिंग परिशुद्धता के लिए देखने के लिए शीर्ष तत्वों की संख्या।
रिटर्न:
कंस्ट्रक्टर और डिस्ट्रक्टर्स | |
---|---|
InTopKV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, :: tensorflow::Input k) |
सार्वजनिक विशेषताएँ | |
---|---|
operation | |
precision |
सार्वजनिक कार्य | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
सार्वजनिक विशेषताएँ
ऑपरेशन
Operation operation
शुद्धता
::tensorflow::Output precision
सार्वजनिक कार्य
InTopKV2
InTopKV2( const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, ::tensorflow::Input k )
नोड
::tensorflow::Node * node() const
ऑपरेटर :: टेंसरफ़्लो :: इनपुट
operator::tensorflow::Input() constहै
ऑपरेटर :: टेंसोफ़्लो :: आउटपुट
operator::tensorflow::Output() const