本文共 129 字，大约阅读时间需要 1 分钟。
import torchinput=torch.Tensor([0.1,0.2]).cuda()k=3v,k=torch.topk(input, k, dim=0, largest=True, sorted=True, out=None)print(v,k)
转载地址：http://qyyk.baihongyu.com/