Tensorflow :: ops :: InTopK

#include <nn_ops.h>

Gibt an, ob die Ziele in den Top- K Vorhersagen enthalten sind.

Zusammenfassung

Dies gibt ein batch_size Array von batch_size . Ein Eintrag out[i] ist true wenn die Vorhersage für die batch_size zu den Top- k Vorhersagen unter allen Vorhersagen gehört, zum Beispiel i . Man beachte , dass das Verhalten von InTopK unterscheidet sich von der TopK op in der Handhabung von Bindungen; Wenn mehrere Klassen den gleichen Vorhersagewert haben und die oberste k Grenze überschreiten, werden alle diese Klassen als oberste k .

Formeller, lassen Sie

\(predictions_i\) sind die Vorhersagen für alle Klassen zum Beispiel i , \(targets_i\) sind die Zielklasse zum Beispiel i , \(out_i\) sind die Ausgabe zum Beispiel i ,

$$out_i = predictions_{i, targets_i} TopKIncludingTies(predictions_i)$$

Argumente:

  • scope: Ein Scope- Objekt
  • Vorhersagen: Ein Tensor für batch_size x- classes .
  • Ziele: Ein batch_size Vektor von Klassen-IDs.
  • k: Anzahl der wichtigsten Elemente, die für die Rechengenauigkeit berücksichtigt werden müssen.

Kehrt zurück:

Konstruktoren und Destruktoren

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

Öffentliche Attribute

operation
precision

Öffentliche Funktionen

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Öffentliche Attribute

Operation

Operation operation

Präzision

::tensorflow::Output precision

Öffentliche Funktionen

InTopK

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

Knoten

::tensorflow::Node * node() const

operator :: tensorflow :: Input

 operator::tensorflow::Input() const

operator :: tensorflow :: Output

 operator::tensorflow::Output() const