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

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部