pytorch人工智能之torch.gather算子用法示例
更新时间：2022年09月19日 16:09:07 作者：ctrlA_ctrlC_ctrlV
这篇文章主要介绍了pytorch人工智能之torch.gather算子用法示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
一、用法：
torch.gather 算子用于返回给定索引/下标的 Tensor 元素，在 pytorch 官网文档中的定义如下：
torch.gather( input, dim, index, *, sparse_grad=False, out=None) → Tensor
其用法等价于：
input.gather( dim, index, *, sparse_grad=False, out=None) → Tensor
其中，input 是目标 Tensor ，即被搜索的 Tensor ；dim 是搜索维度（也是 Tensor ），index 是索引。
返回值类型：Tensor
二、代码示例：
概念看不懂没关系，一看代码便知用法。
a = torch.tensor([1, 5, 3, 6, 8]) b = torch.tensor([3]) # 索引为3 c = a.gather(0, b) # 输出a中第0维索引是3的元素：6 # 等价于 c=torch.gather(a,0,b) print(c) # tensor([6])
a = torch.tensor([[1.3, 2, 3, 4.5, 5],
[2.0, 3, 0.3, 4.1, 2],
[6, 7, 8, 9, 2],
[10, 5, 0, 6, 8]])
b = torch.tensor([[1],
[2],
[3],
[4]])
c = torch.gather(a, 1, b) # 输出a中第1维索引分别是1，2，3，4的元素：2，0.3，9，8
print(c) # tensor([[2.0000],[0.3000],[9.0000],[8.0000]])以上就是pytorch人工智能之torch.gather算子用法示例的详细内容，更多关于pytorch算子torch.gather的资料请关注脚本之家其它相关文章！
相关文章
这篇文章主要介绍了python导出hive数据表的schema实例代码，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下2018-01-01
这篇文章主要介绍了Python在字符串中处理html和xml的方法，文中讲解非常细致，代码帮助大家更好的理解和学习，感兴趣的朋友可以了解下2020-07-07
在本篇内容里小编给大家分享的是一篇关于python3获取控制台输入的数据的具体实例内容，需要的朋友们可以学习下。2020-08-08
今天小编就为大家分享一篇关于Python并行分布式框架Celery详解的文章，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2018-10-10
今天小编就为大家分享一篇python实现自动化上线脚本的示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
Python cookbook（数据结构与算法）在字典中将键映射到多个值上的方法
这篇文章主要介绍了Python在字典中将键映射到多个值上的方法,涉及Python针对字典的相关映射与初始化相关操作技巧,需要的朋友可以参考下2018-02-02
详解django的serializer序列化model几种方法
序列化是将对象状态转换为可保持或传输的格式的过程。这篇文章主要介绍了详解django的serializer序列化model几种方法。具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-10-10
这篇文章主要介绍了python SVM 线性分类模型的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-07-07
今天小编就为大家分享一篇python3实现绘制二维点图，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了python深copy和浅copy区别对比解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-12-12
最新评论