przepływ tensorowy:: ops:: WTopK

#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

InTopK (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, int64 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

WTopK 

 InTopK(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input predictions,
  ::tensorflow::Input targets,
  int64 k
)

węzeł 

::tensorflow::Node * node() const

operator::tensorflow::Wejście 

 operator::tensorflow::Input() const

operator::tensorflow::Wyjście 

 operator::tensorflow::Output() const