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ędzie predykcją dla wszystkich klas, na przykład i , \(targets_i\) będzie klasą docelową, na przykład i , \(out_i\) będzie wynikiem, 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

operacja

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