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 25 Python
约瑟夫问题的Python和C++求解方法
Aug 20 Python
python字符类型的一些方法小结
May 16 Python
谈谈python中GUI的选择
Mar 01 Python
Python解析Excle文件中的数据方法
Oct 23 Python
python实现Flappy Bird源码
Dec 24 Python
Python笔记之工厂模式
Nov 20 Python
Flask中endpoint的理解(小结)
Dec 11 Python
python3实现从kafka获取数据,并解析为json格式,写入到mysql中
Dec 23 Python
Python turtle库的画笔控制说明
Jun 28 Python
python装饰器实现对异常代码出现进行自动监控的实现方法
Sep 15 Python
python批量检查两个对应的txt文件的行数是否一致的实例代码
Oct 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知道与问问的采集插件代码
2010/10/12 PHP
PHP统一页面编码避免乱码问题
2015/04/09 PHP
PHP图像处理类库及演示分享
2015/05/17 PHP
php-redis中的sort排序函数总结
2015/07/08 PHP
jquery获取input表单值的代码
2010/04/19 Javascript
Jquery常用技巧收集整理篇
2010/11/14 Javascript
jquery maxlength使用说明
2011/09/09 Javascript
用js实现小球的自由移动代码
2013/04/22 Javascript
JS实现网页上随滚动条滚动的层效果代码
2015/11/04 Javascript
jQuery循环遍历子节点并获取值的方法
2016/04/14 Javascript
jQuery实现div横向拖拽排序的简单实例
2016/07/13 Javascript
AngularJS基础 ng-mousemove 指令简单示例
2016/08/02 Javascript
纯JS实现出生日期[年月日]下拉菜单效果
2018/06/01 Javascript
vue2.0 element-ui中el-select选择器无法显示选中的内容(解决方法)
2018/08/24 Javascript
JS实现的贪吃蛇游戏完整实例
2019/01/18 Javascript
今天,小程序正式支持 SVG
2019/04/20 Javascript
vue中node_modules中第三方模块的修改使用详解
2019/05/31 Javascript
layui点击弹框页面 表单请求的方法
2019/09/21 Javascript
如何基于原生javaScript生成带图片的二维码
2019/11/21 Javascript
Vue 路由间跳转和新开窗口的方式(query、params)
2019/12/25 Javascript
详解JavaScript的this指向和绑定
2020/09/08 Javascript
Vue开发中常见的套路和技巧总结
2020/11/24 Vue.js
详解Python 2.6 升级至 Python 2.7 的实践心得
2017/04/27 Python
python字典DICT类型合并详解
2017/08/17 Python
python 循环读取txt文档 并转换成csv的方法
2018/10/26 Python
PyCharm中关于安装第三方包的三个建议
2020/09/17 Python
一篇文章带你搞定Ubuntu中打开Pycharm总是卡顿崩溃
2020/11/02 Python
Python urllib request模块发送请求实现过程解析
2020/12/10 Python
python使用matplotlib的savefig保存时图片保存不完整的问题
2021/01/08 Python
CSS Grid布局教程之什么是网格布局
2014/12/30 HTML / CSS
英国计算机产品零售商:Novatech(定制个人电脑、笔记本电脑、工作站和服务器)
2018/01/28 全球购物
法国购买隐形眼镜和眼镜网站:Optical Center
2019/10/08 全球购物
资产评估专业大学生求职信
2013/09/29 职场文书
文秘应聘自荐书范文
2014/02/18 职场文书
政法学院毕业生求职信
2014/02/28 职场文书
讨论nginx location 顺序问题
2022/05/30 Servers