PyTorch Tensor对Tensor索引（Index wrt Tensor）

最新推荐文章于 2024-03-10 11:31:01 发布
VIP文章 最新推荐文章于 2024-03-10 11:31:01 发布
阅读量888 收藏 2
点赞数 1
分类专栏： Python Pytorch 文章标签： pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/HappyCtest/article/details/110289768
版权

简介

模型越编越复杂，有时候就会涉及到一些PyTorch里面没有的Tensor操作，这时候就得自己写。

最近需要实现Tensor对Tensor的索引，即对于一个Tensor中的每个元素，返回它在另一个Tensor中的位置，例子如下：

> inp = torch.LongTensor([5, 3, 6, 9])
> index = torch.LongTensor([2, 3, 4, 6, 8, 5, 7, 9])
> out = tensor_index(inp, index)
> print(out)
tensor([5, 1, 3, 7]
最低0.47元/天 解锁文章
扣德夫特
关注
  • 1
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
Pytorch 中的 tensor index
Do not mess it up
03-25 3380
遇到了用tensor来index另外一个tensor的操作，在Pytorch中右2个比较相似的操作 torch.index_select 和torch.gather torch.index_select(input,dim,index,out=None)→ Tensor Returns a new tensor which indexes theinputtensor a...
Eigen中Tensor的使用技巧
luffytom的博客
04-11 5489
目录：使用Eigen的Tensor缘由Tensor的矩阵乘法操作Tensor的reshape操作Tensor的堆叠(stack)操作Tensor的find（where）操作 使用Eigen的Tensor缘由 之所以使用Eigen的Tensor，主要是因为Tensorflow中的一个C++项目中，需要对graph中得到的Tensor进行数据后处理。而tensorflow的C++API中用到了Eige...
torch中tensor的普通索引以及index_select、torch.gather
qq_43391414的博客
09-27 1万+
今天讲二维和三维数组如何索引，重点在二维，三维是从二维拓展而来。 二维 二维数组很容易索引，例如： c=torch.rand(2,3) print(c) #取行 print(c[[0,1],:]) #取列 print(c[:,[1,2]]) #取行列，所以就是某个元素了。 print(c[[0],[0]])#取(0,0) print(c[[0,0,0],[0,1,2]])#取(0,0),(0,1),(0,2) 需要注意的是，上述写成下述则不是我们想要的： index=torch.tensor(
pytorch每日一学34(torch.index_select())从tensor中按维度取出指定索引的元素
Fluid_ray的博客
12-01 6509
第34个方法 torch.index_select(input, dim, index, *, out=None) → Tensor 此方法的作用是，根据dim取出input中的index对应的元素，并且返回一个tensor，首先是参数介绍 input(Tensor)：输入的tensor。 dim(int)：指定我们要进行索引选择的维度。 index(LongTensor)：索引。 out：输出tensor。 使用方法： >>> x = torch.randn(3, 4) >&
pytorch中的数据索引
最新发布
jakelihua
03-10 876
在PyTorch中，数据索引是指在处理张量（Tensor）时访问或操作特定元素的过程。索引在数据处理和深度学习中是非常常见且重要的操作，它允许我们以各种方式访问数据集中的元素，执行数据的切片、提取、过滤等操作。
PyTorch中Tensor的查找和筛选
热门推荐
玄云飘风的博客
12-29 6万+
本文源码基于版本1.0，交互界面基于0.4.1 import torch 按照指定轴上的坐标进行过滤 index_select() 沿着某tensor的一个轴dim筛选若干个坐标 >>> x = torch.randn(3, 4) # 目标矩阵 >>> x tensor([[ 0.1427, 0.0231, -0.5414, -1.0009], ...
Pytorch tensor 如何找到特定值的索引 (index)
Wu的博客
02-19 2万+
使用 torch.nonzero()，返回非零值的索引 (index) 其中 True 算作非零数，False 算作零，所以可以巧用判断式来找到 Tensor 特定值的索引 import torch a = torch.arange(3*5).reshape(3,5).view(-1) b = torch.nonzero(a==10).squeeze() print(b) # tensor(10) ...
Pytorch Tensor的索引与切片例子
09-18
今天小编就为大家分享一篇Pytorch Tensor的索引与切片例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Pytorch Tensor 输出为txt和mat格式方式
09-18
今天小编就为大家分享一篇Pytorch Tensor 输出为txt和mat格式方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
pytorch 改变tensor尺寸的实现
09-18
今天小编就为大家分享一篇pytorch 改变tensor尺寸的实现，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
pytorch 实现删除tensor中的指定行列
09-18
今天小编就为大家分享一篇pytorch 实现删除tensor中的指定行列，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Tensor的索引和数据筛选
huihuixiaoxue的博客
01-02 325
where
tensor笔记1--index_select函数（对tensor维度的理解）
FUN6367的博客
07-17 1670
pytorch中index_select函数的作用选取某一维度上的数据 函数形式为： index_select(input, dim, index) input为tensor，dim是维度从0开始，index是一维tensor（向量），表示在这个维度上要选择的下标 下边直接上例子： import torch x = torch.Tensor([[[1, 2, 3], [4, 5, 6]], [[9, 8, 7],
关于pytorch用tensor索引另一个tensor问题
ld326的专栏
03-26 3012
1. 问题 在项目中关于如下代码出现问题： def fun1(): start_transitions = torch.nn.Parameter(torch.empty(5)) torch.nn.init.uniform_(start_transitions, -0.1, 0.1) tags_b = tags.byte() rs = start_transition...
Pytorch 基础之张量索引
zxhandroid的博客
02-20 801
主要介绍了一些常用的张量索引 API ，多理解，根据实际情况选择合适的 API 即可，还要注意避免常见的索引越界错误！
pytorch中使用tensor作为另一个tensor的索引 (torch.gather)
xinong123456123的博客
10-20 1334
使用tensor对tensor进行索引的方法
Pytorch：浅探Tensor的各种索引形式
要相信光
12-18 6075
文章目录前因各种Tensor索引操作1. 简单索引2. 一般的花式索引3. 复杂的花式索引4. Informer代码示例 前因 之前一直以为对ndarray的各种索引切片操作还算得上熟悉，但今天师弟问了我Informer实现中ProbSparse Self-Attention的一些Tensor索引操作，才发现有些操作还不太懂，而网上也缺乏相关的参考资料。因此在一系列探索下，写下了这篇博客。 各种Tensor索引操作 构造示例数组x，为一个三维tesnor： import torch x = torch.ar
PyTorch|Tensor基础操作：拼接、切分、索引和变换
木千之
01-11 2470
在学习了Tensor的创建方法之后，接下来你可能会问：那么我们可以对Tensor进行哪些操作呢？不急，今天我们就来聊聊Tensor的操作方法。这部分主要包含两类：Tensor的基础操作：如拼接、切分、索引和变换Tensor的数学运算。
tensor也可以作为索引
Mr.horse的博客
12-03 4222
在TensorFlow中，tensor也可以作为索引，但只能作为同样为tensor类型变量的索引，不能作为list类型变量的索引如下面的例子： import tensorflow as tf index = tf.to_int32([0,1,2]) # index是一个tensor a = [[1,2,3], [4,5,6]] b = a[index[0]] sess = tf.Inter...
pytorch tensor切片
07-15
1.基本切片操作：您可以使用索引操作符[]来对tensor进行切片。例如，如果有一个3x3的tensor，可以使用`tensor[1:3, 0:2]`来获得第二行和第三行的前两列。 2.索引规则：切片操作的索引是从0开始的。在切片时，起始...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值