TensorFlow函数教程:tf.keras.backend.in_top_k
2019-03-13 17:35 更新
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
。