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