tensorflow::ops::InTopK

#include <nn_ops.h>

Says whether the targets are in the top K predictions.

This outputs a batch_size bool array, an entry out[i] is true if the prediction for the target class is among the top k predictions among all predictions for example i. Note that the behavior of InTopK differs from the TopK op in its handling of ties; if multiple classes have the same prediction value and straddle the top-k boundary, all of those classes are considered to be in the top k.

More formally, let

be the predictions for all classes for example i, be the target class for example i, be the output for example i,

Args:

  • scope: A Scope object
  • predictions: A batch_size x classes tensor.
  • targets: A batch_size vector of class ids.
  • k: Number of top elements to look at for computing precision.

Returns:

Constructors and Destructors

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

Public functions

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

Public attributes

operation

Operation operation

precision

::tensorflow::Output precision

Public functions

InTopK

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

node

::tensorflow::Node * node() const 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const