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
,
$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$
Argumenty:
- zakres: Obiekt Scope
- przewidywania:
classes
batch_size
x. - targets: wektor
batch_size
identyfikatoró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