tf.nn.top_k

tf.nn.top_k(
    input,
    k=1,
    sorted=True,
    name=None
)

查找最后一个维度的k个最大项的值和索引。如果输入是一个向量(秩=1),找到向量中k个最大的元素,并将它们的值和索引作为向量输出。因此value [j]是输入的第j个最大的条目,它的索引是index [j]。矩阵(分别地。,计算每一行的前k个条目(resp)。沿着最后一个维度的向量)。因此,

values.shape = indices.shape = input.shape[:-1] + [k]

如果两个元素相等,则首先出现下标元素。

参数:

  • input: 一维或更高张量,最后维数至少为k。
  • k: 0-D int32张量。要沿着最后一个维度查找的顶部元素的数量(对于矩阵,沿着每一行查找)。
  • sorted: 如果为真,则得到的k个元素将按降序排列。
  • name: 操作的可选名称。

返回值:

  • values: 沿最后一个维度切片的k个最大元素。
  • indices: 输入的最后一个维度内的值的索引。

原链接: https://tensorflow.google.cn/versions/r1.9/api_docs/python/tf/nn/top_k?hl=en

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券