torch.topk

torch.kthvalue(input, k, dim=None, keepdim=False, out=None) -> (Tensor, LongTensor)

k：第k个最小元素,返回第k个最小元素

(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

返回前k个最大元素，注意是前k个，largest=False，返回前k个最小元素

此函数的功能是求取1-D 或N-D Tensor的最低维度的前k个最大的值，返回值为两个Tuple，其中values是前k个最大值的Tuple，indices是各个值对应的下标，默认返回结果是从大到小排序的。

使用示例：

topk = tf.nn.top_k(probability_array, 3, name="topk")