参数:
input(Tensor) —- 要排序的张量
dim(int,可选的) —- 沿着此维度排序
descending(bool,可选的) —- 布尔值,控制升序排序
out(tuple,可选的)...如果不指定dim,则默认input的最后一维,如果largest为False,则返回最小的k个值。...参数:
input(Tensor) —- 输入张量
k(int) —- “top-k”中的k值
dim(int,可选的) —- 排序的维度
largest(bool,可选的) —- 布尔值,控制返回最大或最小值...sorted(bool,可选的) —- 布尔值,控制返回值是否排序
out(tuple,可选的) —- 可选输出张量
返回值: 返回一个元组(values, indices),其中indices是原始输入张量...如果设定布尔值sorted为True,将会确保返回的k个值被排序
x = torch.arange(1, 6)
x
tensor([1, 2, 3, 4, 5])
torch.topk