tensoreflusso:: ops:: InTopKV2
#include <nn_ops.h>
Indica se gli obiettivi rientrano nelle previsioni K
principali.
Riepilogo
Ciò restituisce un array bool batch_size
, una voce out[i]
è true
se la previsione per la classe di destinazione è tra le prime k
previsioni tra tutte le previsioni, ad esempio i
. Si noti che il comportamento di InTopK
differisce dall'operazione TopK
nella gestione dei valori di parità; se più classi hanno lo stesso valore di previsione e si trovano a cavallo del limite top- k
, tutte quelle classi sono considerate nel top k
.
Più formalmente, lasciamo
\(predictions_i\) essere le previsioni per tutte le classi, ad esempio i
, \(targets_i\) essere la classe di destinazione, ad esempio i
, \(out_i\) essere l'output per esempio i
,
$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$
Argomenti:
- scope: un oggetto Scope
- previsioni: un tensore
classes
batch_size
x. - target: un vettore
batch_size
di ID di classe. - k: numero di elementi principali da considerare per la precisione del calcolo.
Resi:
Costruttori e distruttori | |
---|---|
InTopKV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, :: tensorflow::Input k) |
Attributi pubblici | |
---|---|
operation | |
precision |
Funzioni pubbliche | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Attributi pubblici
operazione
Operation operation
precisione
::tensorflow::Output precision
Funzioni pubbliche
InTopKV2
InTopKV2( const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, ::tensorflow::Input k )
nodo
::tensorflow::Node * node() const
operatore::tensorflow::Input
operator::tensorflow::Input() const
operatore::tensorflow::Output
operator::tensorflow::Output() const