pytorch索引查找 index_select的例子


Posted in Python onAugust 18, 2019

index_select

anchor_w = self.FloatTensor(self.scaled_anchors).index_select(1, self.LongTensor([0]))

参数说明:index_select(x, 1, indices)

1代表维度1,即列,indices是筛选的索引序号。

例子:

import torch
 
 
x = torch.linspace(1, 12, steps=12).view(3,4)
 
print(x)
indices = torch.LongTensor([0, 2])
y = torch.index_select(x, 0, indices)
print(y)
 
z = torch.index_select(x, 1, indices)
print(z)
 
z = torch.index_select(y, 1, indices)
print(z)

结果:

tensor([[ 1.,  2.,  3.,  4.],
    [ 5.,  6.,  7.,  8.],
    [ 9., 10., 11., 12.]])
tensor([[ 1.,  2.,  3.,  4.],
    [ 9., 10., 11., 12.]])
tensor([[ 1.,  3.],
    [ 5.,  7.],
    [ 9., 11.]])
tensor([[ 1.,  3.],
    [ 9., 11.]])

以上这篇pytorch索引查找 index_select的例子就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
在Python中利用Into包整洁地进行数据迁移的教程
Mar 30 Python
Python的pycurl包用法简介
Nov 13 Python
Python语言实现获取主机名根据端口杀死进程
Mar 31 Python
Python自动化运维和部署项目工具Fabric使用实例
Sep 18 Python
python爬虫之BeautifulSoup 使用select方法详解
Oct 23 Python
Django中日期处理注意事项与自定义时间格式转换详解
Aug 06 Python
Python面向对象之静态属性、类方法与静态方法分析
Aug 24 Python
pycharm中成功运行图片的配置教程
Oct 28 Python
Python中join()函数多种操作代码实例
Jan 13 Python
python中JWT用户认证的实现
May 18 Python
Python工程师必考的6个经典面试题
Jun 28 Python
python中pd.cut()与pd.qcut()的对比及示例
Jun 16 Python
浅谈Pytorch中的torch.gather函数的含义
Aug 18 #Python
PyTorch中Tensor的维度变换实现
Aug 18 #Python
PyTorch中Tensor的拼接与拆分的实现
Aug 18 #Python
详解PyTorch中Tensor的高阶操作
Aug 18 #Python
浅析PyTorch中nn.Linear的使用
Aug 18 #Python
Pytorch实现GoogLeNet的方法
Aug 18 #Python
PyTorch之图像和Tensor填充的实例
Aug 18 #Python
You might like
PHP常用代码
2006/11/23 PHP
生成php程序的php代码
2008/04/07 PHP
PHP MemCached高级缓存配置图文教程
2010/08/05 PHP
PHP如何解决网站大流量与高并发的问题
2011/06/25 PHP
关于二级目录拖拽排序的实现(源码示例下载)
2013/04/26 PHP
又十个超级有用的PHP代码片段
2015/09/24 PHP
PHP数据库操作四:mongodb用法分析
2017/08/16 PHP
asp.net下使用jquery 的ajax+WebService+json 实现无刷新取后台值的实现代码
2010/09/19 Javascript
一个关于jqGrid使用的小例子(行按钮)
2011/11/04 Javascript
通过百度地图获取公交线路的站点坐标的js代码
2012/05/11 Javascript
JS实现向表格行添加新单元格的方法
2015/03/30 Javascript
jquery ui resize 中border-box的bug修正
2015/04/26 Javascript
JS函数定义方式的区别介绍
2016/03/22 Javascript
常用js,css文件统一加载方法(推荐) 并在加载之后调用回调函数
2016/09/23 Javascript
JavaScript如何实现图片懒加载(lazyload) 提高用户体验(增强版)
2016/11/30 Javascript
详谈Ajax请求中的async:false/true的作用(ajax 在外部调用问题)
2017/02/10 Javascript
Angular 表单控件示例代码
2017/06/26 Javascript
jQuery EasyUI window窗口使用实例代码
2017/12/25 jQuery
JavaScript显式数据类型转换详解
2019/03/18 Javascript
Vue实现日历小插件
2019/06/26 Javascript
vue实现配置全局访问路径头(axios)
2019/11/01 Javascript
Vue 数组和对象更新,但是页面没有刷新的解决方式
2019/11/09 Javascript
react国际化化插件react-i18n-auto使用详解
2020/03/31 Javascript
Python编程入门的一些基本知识
2015/05/13 Python
使用python遍历指定城市的一周气温
2017/03/31 Python
python密码错误三次锁定(实例讲解)
2017/11/14 Python
Python实现的查询mysql数据库并通过邮件发送信息功能
2018/05/17 Python
python3中eval函数用法使用简介
2019/08/02 Python
新加坡时尚网上购物:Zalora新加坡
2016/07/26 全球购物
教师自我鉴定范文
2013/11/10 职场文书
会计专业毕业生求职信分享
2014/01/03 职场文书
军训自我鉴定范文
2014/02/13 职场文书
八项规定整改方案
2014/10/01 职场文书
2014年安全生产工作总结
2014/11/13 职场文书
应急管理工作总结2015
2015/05/04 职场文书
天河观后感
2015/06/11 职场文书