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的例子就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持小牛知识库。
-
本文向大家介绍Pytorch Tensor的索引与切片例子，包括了Pytorch Tensor的索引与切片例子的使用技巧和注意事项，需要的朋友参考一下 1. Pytorch风格的索引 根据Tensor的shape，从前往后索引，依次在每个维度上做索引。 示例代码： 上述代码创建了一个shape=[4, 3, 28, 28]的Tensor，我们可以理解为4张图片，每张图片有3个通道，每个通道是28x
-
本文向大家介绍在PyTorch中Tensor的查找和筛选例子，包括了在PyTorch中Tensor的查找和筛选例子的使用技巧和注意事项，需要的朋友参考一下 本文源码基于版本1.0，交互界面基于0.4.1 import torch 按照指定轴上的坐标进行过滤 index_select() 沿着某tensor的一个轴dim筛选若干个坐标 where() 用于将两个broadcastable的tenso
-
索引查找 索引查找的定义，来源百度百科 索引查找是在索引表和主表(即线性表的索引存储结构)上进行的查找。 索引查找的过程是： 首先根据给定的索引值K1，在索引表上查找出索引值等于K1的索引项，以确定K1对应的子表在主表中的开始位置和长度， 然后再根据给定的关键字K2，在对应的子表中查找出关键字等于K2的元素(结点)。 思路说明 对于一个list或者dictionary类型的数据，python有专门
-
问题内容： 老实说，这个问题很简单。有没有一种方法可以快速查找数组中某个元素的所有出现而无需循环遍历它？似乎所有内置方法仅返回第一次出现的索引，而不是全部返回。 拥有返回索引数组的样式方法将是很不错的。有什么想法吗？ 预先感谢您的任何投入！ 编辑： 感谢大家的回复！看来我应该对此更加清楚。我目前的操作方式是扩展，其外观与下面发布的一个哑光非常相似。我知道，任何执行此操作的方法都必须在内部遍历数组，
-
问题内容： 我正在为我正在写的插件编写迁移生成器，我需要能够找到表具有的唯一索引，以便可以将现有的唯一索引修改为复合唯一索引。我一直在尝试找到一种方法来访问表具有ActiveRecord的索引。我只能找到ActiveRecord :: ConnectionAdapters :: PostgreSQLAdapter :: indexes方法，但是不幸的是，这仅适用于PosgreSQLAdapter。
-
问题内容： 例如我有这个数组： 如何从对象获取价值指数？例如“ Country”：“ Austria”，该指数为3 问题答案： 您可以在最新的浏览器中使用，但Internet Explorer中不支持此功能，只有Edge 为了在IE9及更高版本中提供支持，您可以改用