tensorflow :: ops :: InTopKV2

#include <nn_ops.h>

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

Resumo

Isso gera um batch_size 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 do TopK op no tratamento de empates; se várias classes tiverem o mesmo valor de predição e ultrapassarem o limite superior k , todas essas classes serão consideradas no topo k .

Mais formalmente, vamos

\(predictions_i\) ser as previsões para todas as classes, por exemplo i , \(targets_i\) ser a classe de destino, por exemplo i , \(out_i\) ser a saída, por exemplo i

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

Argumentos:

  • escopo: um objeto Scope
  • previsões: Um tensor batch_size x classes .
  • alvos: um vetor batch_size de ids de classe.
  • k: Número de elementos principais a serem observados para precisão de computação.

Retorna:

Construtores e Destruidores

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

InTopKV2

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

nó

::tensorflow::Node * node() const

operador :: tensorflow :: Input

 operator::tensorflow::Input() const

operador :: tensorflow :: Saída

 operator::tensorflow::Output() const