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字符串对其居中显示的方法
Jul 11 Python
python实现SMTP邮件发送功能
Jun 16 Python
python中安装模块包版本冲突问题的解决
May 02 Python
Python实现的rsa加密算法详解
Jan 24 Python
python中kmeans聚类实现代码
Feb 23 Python
华为校园招聘上机笔试题 扑克牌大小(python)
Apr 22 Python
使用django实现一个代码发布系统
Jul 18 Python
Django REST Framework序列化外键获取外键的值方法
Jul 26 Python
40行Python代码实现天气预报和每日鸡汤推送功能
Feb 27 Python
pycharm实现在虚拟环境中引入别人的项目
Mar 09 Python
Django 用户认证Auth组件的使用
Nov 30 Python
Python日志模块logging用法
Jun 05 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编写发红包程序
2015/07/22 PHP
php实现的错误处理封装类实例
2017/06/20 PHP
解决Laravel自定义类引入和命名空间的问题
2019/10/15 PHP
Centos7安装swoole扩展操作示例
2020/03/26 PHP
javascript 读取xml,写入xml 实现代码
2009/07/10 Javascript
javascript 正则替换 replace(regExp, function)用法
2010/05/22 Javascript
boxy基于jquery的弹出层对话框插件扩展应用 弹出层选择器
2010/11/21 Javascript
jquery下checked取值问题的解决方法
2012/08/09 Javascript
打造个性化的功能强大的Jquery虚拟键盘(VirtualKeyboard)
2014/10/11 Javascript
JS实现让访问者自助选择网页文字颜色的方法
2015/02/24 Javascript
学习javascript的闭包,原型,和匿名函数之旅
2015/10/18 Javascript
以Python代码实例展示kNN算法的实际运用
2015/10/26 Javascript
基于JavaScript实现百叶窗动画效果不只单纯flas可以实现
2016/02/29 Javascript
Vue.js每天必学之过滤器与自定义过滤器
2016/09/07 Javascript
jQuery插件FusionCharts实现的3D帕累托图效果示例【附demo源码】
2017/03/25 jQuery
JavaScript find()方法及返回数据实例
2020/04/30 Javascript
如何通过vscode运行调试javascript代码
2020/07/24 Javascript
JS实现悬浮球只在一侧滑动并且是横屏状态下
2020/08/19 Javascript
VUE+Element实现增删改查的示例源码
2020/11/23 Vue.js
Python按行读取文件的实现方法【小文件和大文件读取】
2016/09/19 Python
Python统计一个字符串中每个字符出现了多少次的方法【字符串转换为列表再统计】
2019/05/05 Python
使用Python测试Ping主机IP和某端口是否开放的实例
2019/12/17 Python
利用Tensorflow构建和训练自己的CNN来做简单的验证码识别方式
2020/01/20 Python
HTML5注册页面示例代码
2014/03/27 HTML / CSS
Lee牛仔裤澳大利亚官网:美国著名牛仔裤品牌
2017/09/02 全球购物
哪些情况下不应该使用索引
2015/07/20 面试题
什么是lambda函数
2013/09/17 面试题
大学毕业生求职自荐书
2014/06/05 职场文书
2014党员学习习主席讲话思想汇报
2014/09/15 职场文书
2014年党员干部四风问题自我剖析材料
2014/09/29 职场文书
招标保密承诺书
2015/01/20 职场文书
2016年秋季运动会通讯稿
2015/11/25 职场文书
小学五年级(说明文3篇)
2019/08/13 职场文书
JavaScript 数组去重详解
2021/09/15 Javascript
vue3种table表格选项个数的控制方法
2022/04/14 Vue.js
CSS SandBox应用场景及常见问题
2022/06/25 HTML / CSS