torch 多维topk

flatten和view(-1) 是按行展开，一行一行的拼接

    import torchanch_ious = torch.Tensor([[1, 2, 3,4], [7,8,9,3]])# anch_ious = torch.Tensor([[1, 2, 3,4], [4, 5, 6,7]]).view(-1)neg_count=3top_data,index= torch.topk(anch_ious, neg_count, dim=1, largest=True, sorted=True, out=None)# print(anch_ious[index])index_num = torch.arange(0, anch_ious.size(0)).repeat(3,1)index_num=torch.flatten(index_num.t())print(index_num)# indexs = index[:, 1:]index = torch.flatten(index)print(anch_ious[index_num,index])

torch 多维topk相关推荐

torch中的topk()函数
torch中的topk()函数 In [2]: import torchIn [3]: a=torch.randn((4,6))In [4]: a Out
Torch 二维多通道卷积运算方式
全连接卷积对于单通道卷积的运算,相信大家已经见过不少了那么在卷积神经网络中,图像的通道数是怎样实现"从 3 到 8" 这样的跳变呢? 接下来以尺寸 5 × 5 的卷积核做一下实 ...
torch多维取数据
索引类型只支持uint8或者int64 bytes 或者long类型 import torch import numpy as np a=torch.zeros((4,3,3))b=torch.one ...
numpy实现torch的topk方法
torch中提供了topk方法用来返回矩阵中对应维度中最大的K个元素以及在对应维度中的index,但是numpy并没有提供和torch一样的topk方法,所以在这里通过numpy的argpartiti ...
torch对于tensor的常规操作
前言使用pytorch框架,会常操作tensor,以下则是对tensor常规操作的汇总. import torch torch.Tensor会继承某些torch的某些数学运算,例如sort, min ...
06_2_Pytorch的基础数据类型、CPU tensor类型和GPU tensor类型、判断数据类型、CPU或GPU张量之间的转换、数据类型转换、1-4维向量、Tensor张量、Variable等
1.6.2.Pytorch的基础数据类型 1.6.2.1.Torch定义了的七种CPU tensor类型和八种GPU tensor类型 1.6.2.2.基础数据类型 1.6.2.3.Pytorch数据 ...
pytorch基础知识+构建LeNet对Cifar10进行训练+PyTorch-OpCounter统计模型大小和参数量+模型存储与调用
整个环境的配置请参考我另一篇博客.ubuntu安装python3.5+pycharm+anaconda+opencv+docker+nvidia-docker+tensorflow+pytorch+C ...
机器学习花朵图像分类_在PyTorch中使用转移学习进行图像分类
想了解更多好玩的人工智能应用,请关注公众号"机器AI学习数据AI挖掘","智能应用"菜单中包括:颜值检测.植物花卉识别.文字识别.人脸美妆等有趣的智能应用.. ...
2019.7学习总结-目标检测-Python+pytorch
目录 1.基础知识 1.1.dict的copy: 1.2.torch.cuda.synchronize() 1.3.在CenterNet里 1.4.torch.gather.topk用法: 1.5.t ...

torch 多维topk

torch 多维topk相关推荐

最新文章

热门文章