torch.topk

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_34690929/article/details/80029800

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

博主设置当前文章不允许评论。

没有更多推荐了,返回首页