fluxo tensor:: ops:: InTopK

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:

• Output : Precisão calculada em k como um bool Tensor .

Atributos públicos

Operation operation

::tensorflow::Output precision

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const