TensorFlow函數(shù)教程：tf.keras.backend.in_top_k

tf.keras.backend.in_top_k函數(shù)

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

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

判斷targets是否在predictions的前k個(gè)中。

參數(shù)：

• predictions：一個(gè)張量，shape為(batch_size, classes)，類型為float32。
• targets：一個(gè)1D張量，長度為batch_size，類型為int32或int64。
• k：一個(gè)int，要考慮的頂級元素的數(shù)量。

返回：

一個(gè) 1D 張量，長度為batch_size，類型為bool。如果predictions[i, targets[i]]在predictions[i]的 top-k值中，則output[i]為True。