今天小编就为大家分享一篇PyTorch中topk函数的用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

听名字就知道这个函数是用来求tensor中某个dim的前k大或者前k小的值以及对应的index。 用法 torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor) input：一个tensor数据 k：指明是得到前k个数据以及其index dim： 指定在哪个维度上排序， 默认是最后一个维度 largest：如果为True，按照大到小排序； 如果为False，按照小