tensorflow::
ops::
InTopK
#include <nn_ops.h>
Says whether the targets are in the top
K
predictions.
Summary
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
\(predictions_i\) be the predictions for all classes for example
i
, \(targets_i\) be the target class for example
i
, \(out_i\) be the output for example
i
,
Args:
- scope: A Scope object
-
predictions: A
batch_size
xclasses
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 attributes |
|
---|---|
operation
|
|
precision
|
Public functions |
|
---|---|
node
() const
|
::tensorflow::Node *
|
operator::tensorflow::Input
() const
|
|
operator::tensorflow::Output
() const
|
|
Public attributes
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