tf.keras.backend.in_top_k(
predictions, targets, k
)
Аргументы
predictions
Массив тензорного типа float32, с формой (batch_size, classes).
targets
Одномерный тензор длиной batch_size типа int32 или int64.
k
Целое число, количество лучших элементов для рассмотрения.
Возвращает
Одномерный тензор длиной batch_size и типа bool. output[i] имеет значение True, если predictions[i, targets[i]] находится среди лучших k значений predictions[i].