pytorch索引查找 index_select的例子
更新时间：2019年08月18日 19:18:23 作者：ShellCollector
今天小编就为大家分享一篇pytorch索引查找 index_select的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
登录验证码是每个网站登录时的基本标配，这篇文章主要介绍了Django实现登录随机验证码的示例代码，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-06-06
下面小编就为大家带来一篇Python爬虫之xlml解析库(全面了解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-08-08
这篇文章主要介绍了python matplotlib的相关资料，帮助大家更好的利用python matplotlib库绘制图表，感兴趣的朋友可以了解下2020-10-10
这篇文章主要介绍了python如何导出微信公众号文章方法详解,本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-08-08
这篇文章主要介绍了pyspark对Mysql数据库进行读写的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-12-12
本文给大家汇总介绍下使用Python实现身份证号码验证解析的几个方法，有需要的小伙伴可以参考下。2015-09-09
这篇文章主要介绍了python中select模块的相关资料，Python中的select模块专注于I/O多路复用，提供了select、poll和epoll三个方法，文中还详细的介绍了select和poll，因为poll与select相差不大，所以本文不作介绍，需要的朋友们下面来一起看看吧。2017-04-04
本文主要介绍了python实现tree命令的使用示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2023-09-09
这篇文章主要介绍了使用pycharm连接读取orcl数据库的表的操作方法,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2024-01-01
这篇文章主要介绍了Python Selenium实现无可视化界面过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-08-08
最新评论