tf.raw_ops.InTopK

Says whether the targets are in the top K predictions.

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.raw_ops.InTopK

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

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

outi=predictionsi,targetsiTopKIncludingTies(predictionsi)

predictions A Tensor of type float32. A batch_size x classes tensor.
targets A Tensor. Must be one of the following types: int32, int64. A batch_size vector of class ids.
k An int. Number of top elements to look at for computing precision.
name A name for the operation (optional).

A Tensor of type bool.