przepływ tensorowy:: ops:: WTopKV2
#include <nn_ops.h> Mówi, czy cele znajdują się w górnych przewidywaniach K
Streszczenie
Daje to w wyniku tablicę bool batch_size , wpis out[i] jest true jeśli predykcja dla klasy docelowej znajduje się wśród k najlepszych predykcji spośród wszystkich predykcji, na przykład i . Należy pamiętać, że zachowanie InTopK różni się od operacji TopK pod względem obsługi powiązań; jeśli wiele klas ma tę samą wartość przewidywania i leży poza górną granicą k , wszystkie te klasy są uważane za znajdujące się w górnym k .
Bardziej formalnie niech
\(predictions_i\) będą przewidywaniami dla wszystkich klas, na przykład i , \(targets_i\) być klasą docelową, na przykład i , \(out_i\) być wyjściem na przykład i ,
Argumenty:
- zakres: Obiekt Scope
- przewidywania:
classesbatch_sizex. - targets: wektor
batch_sizeidentyfikatorów klas. - k: Liczba najważniejszych elementów, na które należy zwrócić uwagę pod kątem precyzji obliczeń.
Zwroty:
Konstruktory i destruktory | |
|---|---|
InTopKV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, :: tensorflow::Input k) |
Atrybuty publiczne | |
|---|---|
operation | |
precision | |
Funkcje publiczne | |
|---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const | |
Atrybuty publiczne
działanie
Operation operation
precyzja
::tensorflow::Output precision
Funkcje publiczne
WTopKV2
InTopKV2( const ::tensorflow::Scope & scope, ::tensorflow::Input predictions, ::tensorflow::Input targets, ::tensorflow::Input k )
węzeł
::tensorflow::Node * node() const
operator::tensorflow::Wejście
operator::tensorflow::Input() const
operator::tensorflow::Wyjście
operator::tensorflow::Output() const