torch topk

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

pytorch中文官网文档：http://www.mamicode.com/info-detail-2217311.html

沿给定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

1维的可以直接取值，

    import torchanch_ious = torch.Tensor([[1, 2, 3], [4, 5, 6]]).view(-1)neg_count=4top_data,index= torch.topk(anch_ious, neg_count

torch topk相关推荐

PyTorch 笔记（08）— Tensor 比较运算（torch.gt、lt、ge、le、eq、ne、torch.topk、torch.sort、torch.max、torch.min）
1. 常用函数比较函数中有一些是逐元素比较,操作类似逐元素操作,还有一些类似归并操作,常用的比较函数如下表所示. 表中第一行的比较操作已经实现了运算符重载,因此可以使用 a>=b,a>b ...
torch.topk
import torchinput=torch.Tensor([0.1,0.2]).cuda()k=3v,k=torch.topk(input, k, dim=0, largest=True, sor ...
PyTorch搜索Tensor指定维度的前K大个（K小个）元素--------(torch.topk)命令参数详解及举例
torch.topk 语法 torch.topk(input, k, dim=None, largest=True, sorted=True, *, out = None) 作用返回输入tensor ...
Pytorch torch.topk()的简单用法
官方文档:https://pytorch.org/docs/stable/generated/torch.topk.html?highlight=topk#torch.topk 由于numpy本身是没 ...
torch.topk和tensor.topk
https://pytorch.org/docs/stable/generated/torch.topk.html#torch.topk 1. 定义按照指定的维度进行数值大小的排序,返回top-k个 ...
详解torch.topk
torch.tok() torch.topk(input, k, dim=None, largest=True, sorted=True, *, out=None) -> (Tensor, Lo ...
torch.topk() 函数详解
作用: 返回列表中最大的n个值例子1:m=torch.arange(0,10) print(m.topk(3)) torch.return_types.topk( val ...
torch 多维topk
flatten和view(-1) 是按行展开,一行一行的拼接 import torchanch_ious = torch.Tensor([[1, 2, 3,4], [7,8,9,3]])# anch_ ...
numpy实现torch的topk方法
torch中提供了topk方法用来返回矩阵中对应维度中最大的K个元素以及在对应维度中的index,但是numpy并没有提供和torch一样的topk方法,所以在这里通过numpy的argpartiti ...

torch topk

torch topk相关推荐

最新文章

热门文章