pytorch多维筛选

多级筛选：

比如结构是2*2*3，只想选第三维的最大的

tx[index, best_n, g_y_center, g_x_center]

index=[01],best_n=[0,1]

最后只取两个值，第一行，第1列，第二行，第2列的。

筛选第3维最大的值，下面的代码不对，解决方法：查询max源码

也可以把3维用view降到2维再计算就可以了。

    import torchanch_ious = torch.Tensor([[[1, 2, 3]], [[4, 5, 6]], [[7, 8, 9]]])print('b shape',anch_ious.shape)b = torch.max(anch_ious, 2)print(b[0])print(b[1])b = b[1].squeeze(1)print(b)print(anch_ious[list(range(anch_ious.size(0))),list(range(anch_ious.size(1))), b])

通过值筛选：

import torchx = torch.linspace(1, 8, steps=8).view(4, 2)#筛选第一维和第二维都>5.5的
print(x)
area=(x[:,0]>5.5)&(x[:,1]>5.5)
b=x[area]
# b= x[torch.where((x[:,0]&g

pytorch多维筛选相关推荐

pytorch单维筛选相乘
m需要和筛选的结果维度相同 >0.5运行的结果与原来维度相同,结果是 0 1,0代不符合,1代表符合. import torchm=torch.Tensor([0.1,0.2,0.3]).cud ...
pytorch 按维取数据
还是得按照这样取数据准,np.array是不准确的 import numpy as npimport torch#下面这个是错误的: x = torch.ones(2,2,requires_grad= ...
Pytorch二维卷积 conv2d 使用/源码/手写实现conv2d/手写向量内积实现conv2d/转置卷积实现——学习笔记
这里是一份学习笔记- 学习视频点这里
python列表生成器二维筛选_如何加速Eratosthenes python列表生成器的筛选
编辑:已经意识到在SO上有很多优化,但很少有其他人为主筛算法解释它们,因此很难让初学者或第一次算法创建者接近它们.所有解决方案都将在python中,以便在速度和优化的同一页面上.这些解决方案将逐渐变得 ...
有bug！PyTorch在AMD CPU的计算机上卡死了
视学算法报道转载自:机器之心编辑:小舟.陈萍 AMD,No?PyTorch在AMD CPU的机器上出现死锁了. PyTorch 作为机器学习中广泛使用的开源框架,具有速度快.效率高等特点.而近年来 ...
Pytorch-学习记录卷积操作——Tensor.size()
设计网络结构.看网络结构源码时时常需要计算输入在结构中经过卷积层.pooling层后尺寸大小,特此记录. pytorch中二维卷积层接口为 Conv2d() ,参数详情不多赘述,详情请参阅官方文档.我 ...
成为指标的“绝地武士”：tableau创建指标的 10 个技巧和窍门
在 Tableau 2020.2 中,我们引入了指标功能来帮助您在同一个地方以一致的格式跟踪最重要的 KPI .今天,我们将分享一些从现有仪表盘和新仪表盘创建指标的技巧,赶快学起来吧~ 技巧 1:在 ...
神策用户画像 2 大功能：千人千面，一人千面，一次给你
你是否,在纠结,用户为什么对你爱答不理? 你是否,在哀怨,用户为什么只薅羊毛? 你是否,在茫然,用户为什么选 Ta,不选你? 你是否,在后悔,营销花钱如流水,用户却寥寥? 得用户者得天下,懂用户者得未 ...
GIAC | 大数据分析系统在游戏领域的迭代与实践
导语:6月23日,腾讯游戏数据分析系统负责人周东祥在 "GIAC全球互联网架构大会" 的分享了主题为<大数据分析系统在游戏领域的迭代与实践>的内容,具体的分享视频和PP ...

pytorch多维筛选

pytorch多维筛选相关推荐

最新文章

热门文章