import torch input=torch.Tensor([0.1,0.2]).cuda() k=3 v,k=torch.topk(input, k, dim=0, largest=True, sorted=True, out=None) print(v,k)
import torch input=torch.Tensor([0.1,0.2]).cuda() k=3 v,k=torch.topk(input, k, dim=0, largest=True, sorted=True, out=None) print(v,k)