fluxo tensor:: ops:: InTopK

#include <nn_ops.h>

Diz se os alvos estão nas K principais previsões.

Resumo

Isso gera uma matriz bool batch_size , uma entrada out[i] é true se a previsão para a classe de destino estiver entre as k principais previsões entre todas as previsões, por exemplo i . Observe que o comportamento do InTopK difere da operação TopK no tratamento de empates; se múltiplas classes tiverem o mesmo valor de predição e ultrapassarem o limite k superior, todas essas classes serão consideradas no topo k .

Mais formalmente, deixe

\(predictions_i\) sejam as previsões para todas as classes, por exemplo i , \(targets_i\) seja a classe alvo, por exemplo i , \(out_i\) seja a saída, por exemplo, i ,

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

Argumentos:

  • escopo: um objeto Escopo
  • previsões: um tensor classes batch_size x.
  • alvos: um vetor batch_size de IDs de classe.
  • k: Número de elementos principais a serem observados para precisão computacional.

Retorna:

Construtores e Destruidores

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

Atributos públicos

operation
precision

Funções públicas

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

Atributos públicos

operação 

Operation operation

precisão 

::tensorflow::Output precision

Funções públicas

InTopK 

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

nó 

::tensorflow::Node * node() const

operador::tensorflow::Input 

 operator::tensorflow::Input() const

operador::tensorflow::Saída 

 operator::tensorflow::Output() const