TensorFlow函数教程：tf.keras.backend.in_top_k

tf.keras.backend.in_top_k函数

tf.keras.backend.in_top_k(
    predictions,
    targets,
    k
)

定义在：tensorflow/python/keras/backend.py。

判断targets是否在predictions的前k个中。

参数：

• predictions：一个张量，shape为(batch_size, classes)，类型为float32。
• targets：一个1D张量，长度为batch_size，类型为int32或int64。
• k：一个int，要考虑的顶级元素的数量。

返回：

一个 1D 张量，长度为batch_size，类型为bool。如果predictions[i, targets[i]]在predictions[i]的 top-k值中，则output[i]为True。