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的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章
本文主要介绍了Appium自动化测试实现H5页面元素定位，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-02-02
这篇文章给大家介绍了在 PyTorch 中,将数据从 CPU 移动到 GPU 的几种方法,使用 .to() 方法,使用 .cuda() 方法,使用 torch.Tensor 构造函数和使用 torch.tensor 构造函数这四个方法,通过代码示例介绍非常详细,需要的朋友可以参考下2024-01-01
这篇文章主要为大家详细介绍了python+logging+yaml实现日志分割，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-07-07
这篇文章主要介绍了Python xlwt模块使用代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-06-06
Python socket套接字实现C/S模式远程命令执行功能案例
这篇文章主要介绍了Python socket套接字实现C/S模式远程命令执行功能,涉及Python socket套接字编写服务器/客户机模式数据传输相关操作技巧,需要的朋友可以参考下2018-07-07
python使用xlsxwriter实现有向无环图到Excel的转换
这篇文章主要为大家详细介绍了python使用xlsxwriter实现有向无环图到Excel的转换，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-12-12
tensorflow使用range_input_producer多线程读取数据实例
今天小编就为大家分享一篇tensorflow使用range_input_producer多线程读取数据实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
这篇文章介绍了Python实现猜数字小游戏，文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以收藏下，方便下次浏览观看2021-12-12
今天小编就为大家分享一篇python实现智能语音天气预报，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
本文主要介绍了Pytorch从0实现Transformer的实践，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-05-05
最新评论