fluxo tensor:: ops:: InTopKV2
#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
sejam as previsões para todas as classes, por exemplo i
, seja a classe alvo, por exemplo 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 | |
---|---|
InTopKV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input predictions, :: tensorflow::Input targets, :: tensorflow::Input k) |
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