flux tensoriel : : opérations : : DansTopKV2

#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\) soit les prédictions pour toutes les classes par exemple i , \(targets_i\) soit 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.

Retour:

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

Constructeurs et Destructeurs

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

DansTopKV2

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

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: Entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: Sortie

 operator::tensorflow::Output() const