flux tensoriel : : opérations : : DansTopK

#include <nn_ops.h>

Indique si les cibles figurent dans les K premières prédictions.

Résumé

Cela génère un tableau booléen batch_size , une entrée out[i] est true si la prédiction pour la classe cible fait partie des k meilleures prédictions parmi toutes les prédictions, par exemple i . Notez que le comportement d' InTopK diffère de l'opération TopK dans sa gestion des liens ; si plusieurs classes ont la même valeur de prédiction et chevauchent la limite top- k , toutes ces classes sont considérées comme faisant partie du top k .

Plus formellement, disons

\(predictions_i\) être les prédictions pour toutes les classes par exemple i , \(targets_i\) être la classe cible par exemple i , \(out_i\) soit la sortie par exemple i ,

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

Arguments :

  • scope : un objet Scope
  • prédictions : un tenseur batch_size x classes .
  • cibles : un vecteur batch_size d’identifiants de classe.
  • k : nombre d'éléments principaux à prendre en compte pour la précision du calcul.

Retours :

  • Output : Précision calculée à k sous forme de bool Tensor .

Constructeurs et Destructeurs

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

Attributs publics

operation
precision

Fonctions publiques

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

Attributs publics

opération 

Operation operation

précision 

::tensorflow::Output precision

Fonctions publiques

DansTopK 

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

nœud 

::tensorflow::Node * node() const

opérateur :: tensorflow :: Entrée 

 operator::tensorflow::Input() const

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const