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实现按行切分文本文件的方法
这篇文章主要介绍了python实现按行切分文本文件的方法,涉及Python利用shell命令操作文本文件的相关技巧,需要的朋友可以参考下
2016-04-04
Python爬虫之Selenium中frame/iframe表单嵌套页面
这篇文章主要介绍了Python爬虫之Selenium中frame/iframe表单嵌套页面，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-12-12
Python中map,reduce,filter和sorted函数的使用方法
这篇文章主要介绍了Python中map,reduce,filter和sorted函数的使用方法,是Python入门学习中的基础知识,需要的朋友可以参考下
2015-08-08
python 线性回归分析模型检验标准--拟合优度详解
今天小编就为大家分享一篇python 线性回归分析模型检验标准--拟合优度详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-02-02
TensorFlow和Numpy矩阵操作中axis理解及axis=-1的解释
在调用numpy库中的concatenate()时,有遇到axis=-1/1/0的情况,下面这篇文章主要给大家介绍了关于TensorFlow和Numpy矩阵操作中axis理解及axis=-1解释的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
2022-03-03
pandas 读取各种格式文件的方法
今天小编就为大家分享一篇pandas 读取各种格式文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-06-06
python matplotlib折线图样式实现过程
这篇文章主要介绍了python matplotlib折线图样式实现过程,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-11-11
Python networkx包的实现
这篇文章主要介绍了Python networkx包的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-02-02
python解释器spython使用及原理解析
这篇文章主要介绍了python解释器spython使用及原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-08-08
基于Python的图像数据增强Data Augmentation解析
这篇文章主要介绍了基于Python的图像数据增强Data Augmentation解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-08-08