fluxo tensor:: ops:: InTopK
#include <nn_ops.h>
Diz se os alvos estão nas principais previsões K
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
,
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