PyTorch搜索Tensor指定维度的前K大个(K小个)元素--------(torch.topk)命令参数详解及举例
torch.topk语法torch.topk(input,k,dim=None,largest=True,sorted=True,*,out=None)作用返回输入tensorinput中,在给定的维度dim上k个最大的元素。如果dim没有给定,那么选择输入input的最后一维。如果largest=False,那么返回k个最小的元素。返回一个namedtuple类型的元组(values,indic