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中的异常处理
Apr 28 Python
运动检测ViBe算法python实现代码
Jan 09 Python
python3使用smtplib实现发送邮件功能
May 22 Python
对pytorch网络层结构的数组化详解
Dec 08 Python
Python 支付整合开发包的实现
Jan 23 Python
使用python实现ftp的文件读写方法
Jul 02 Python
django做form表单的数据验证过程详解
Jul 26 Python
通过selenium抓取某东的TT购买记录并分析趋势过程解析
Aug 15 Python
Django上线部署之IIS的配置方法
Aug 22 Python
Django连接数据库并实现读写分离过程解析
Nov 13 Python
pytorch梯度剪裁方式
Feb 04 Python
Python Matplotlib绘图基础知识代码解析
Aug 31 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 会话(session 时间设定)使用入门代码
2008/06/05 PHP
Yii实现的多级联动下拉菜单
2016/07/13 PHP
PHP经典实用正则表达式小结
2017/05/04 PHP
PHP使用ActiveMQ实现消息队列的方法详解
2019/05/31 PHP
JS图片浏览组件PhotoLook的公开属性方法介绍和进阶实例代码
2010/11/09 Javascript
js中escape对应的C#解码函数 UrlDecode
2012/12/16 Javascript
select标签模拟/美化方法采用JS外挂式插件
2013/04/01 Javascript
Node.js生成HttpStatusCode辅助类发布到npm
2013/04/09 Javascript
JavaScript strike方法入门实例(给字符串加上删除线)
2014/10/17 Javascript
总结jQuery插件开发中的一些要点
2016/05/16 Javascript
微信小程序 倒计时组件实现代码
2016/10/24 Javascript
使用Node.js给图片加水印的方法
2016/11/15 Javascript
教大家轻松制作Bootstrap漂亮表格(table)
2016/12/13 Javascript
ionic cordova一次上传多张图片(类似input file提交表单)的实现方法
2016/12/16 Javascript
jQuery网页定位导航特效实现方法
2016/12/19 Javascript
基于百度地图api清除指定覆盖物(Overlay)的方法
2018/01/26 Javascript
在vscode中统一vue编码风格的方法
2018/02/22 Javascript
vue.js input框之间赋值方法
2018/08/24 Javascript
vue实现文字加密功能
2019/09/27 Javascript
vue+swiper实现左右滑动的测试题功能
2020/10/30 Javascript
[00:32]2018DOTA2亚洲邀请赛VGJ.T出场
2018/04/03 DOTA
老生常谈Python基础之字符编码
2017/06/14 Python
python3 字符串/列表/元组(str/list/tuple)相互转换方法及join()函数的使用
2019/04/03 Python
python3 re返回形式总结
2020/11/20 Python
python实现跨年表白神器--你值得拥有
2021/01/04 Python
米兰网婚纱礼服法国网上商店:Milanoo法国
2016/08/20 全球购物
英国领先的珍珠首饰品牌:Orchira
2016/09/11 全球购物
芬兰攀岩、山地运动和户外活动用品购物网站:Bergfreunde
2016/10/06 全球购物
高一生物教学反思
2014/01/17 职场文书
学前教育专业求职信
2014/09/02 职场文书
财务会计实训报告
2014/11/05 职场文书
小学少先队活动总结
2015/05/08 职场文书
2015年高三班主任工作总结
2015/05/21 职场文书
繁星春水读书笔记
2015/06/30 职场文书
门面租赁合同范文
2019/08/06 职场文书
MySQL 中如何归档数据的实现方法
2022/03/16 SQL Server