torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)
沿给定dim
维度返回输入张量input
中 k
个最大值。
如果不指定dim
,则默认为input
的最后一维。
如果为largest
为 False ,则返回最小的 k
个值。
返回一个元组 (values,indices)
,其中indices
是原始输入张量input
中测元素下标。
如果设定布尔值sorted
为_True_
,将会确保返回的 k
个值被排序。
参数:
input (Tensor)
– 输入张量
k (int)
– “top-k”中的k
dim (int, optional)
– 排序的维
largest (bool, optional)
– 布尔值,控制返回最大或最小值
sorted (bool, optional)
– 布尔值,控制返回值是否排序
out (tuple, optional)
– 可选输出张量 (Tensor, LongTensor) output buffer