flujo tensor:: operaciones:: EnTopK

#include <nn_ops.h>

Dice si los objetivos están entre las K predicciones principales.

Resumen

Esto genera una matriz bool batch_size , una entrada out[i] es true si la predicción para la clase objetivo se encuentra entre las k predicciones principales entre todas las predicciones, por ejemplo, i . Tenga en cuenta que el comportamiento de InTopK difiere del de la operación TopK en su manejo de los empates; Si varias clases tienen el mismo valor de predicción y se encuentran a ambos lados del límite k superior, se considera que todas esas clases están en el k superior.

Más formalmente, dejemos

\(predictions_i\) sean las predicciones para todas las clases, por ejemplo i , \(targets_i\) sea la clase objetivo, por ejemplo i , \(out_i\) sea la salida, por ejemplo i ,

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

Argumentos:

  • alcance: un objeto de alcance
  • predicciones: un tensor de classes x batch_size .
  • objetivos: un vector batch_size de identificadores de clase.
  • k: Número de elementos superiores a considerar para la precisión informática.

Devoluciones:

Constructores y destructores

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

Atributos públicos

operation
precision

Funciones públicas

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

Atributos públicos

operación

Operation operation

precisión

::tensorflow::Output precision

Funciones públicas

EnTopK

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

nodo

::tensorflow::Node * node() const 

operador::tensorflow::Entrada

 operator::tensorflow::Input() const 

operador::tensorflow::Salida

 operator::tensorflow::Output() const