代码语言：javascript复制
tf.nn.top_k(
input,
k=1,
sorted=True,
name=None
)查找最后一个维度的k个最大项的值和索引。如果输入是一个向量(秩=1)，找到向量中k个最大的元素，并将它们的值和索引作为向量输出。因此value [j]是输入的第j个最大的条目，它的索引是index [j]。矩阵(分别地。，计算每一行的前k个条目(resp)。沿着最后一个维度的向量)。因此,
代码语言：javascript复制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
