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")