pytorch学习笔记 -- gather函数和scatter

pytorch学习笔记 -- gather函数和scatter_()的解析

PyTorch

最后更新 2021-08-19 09:44 阅读 7335

最后更新 2021-08-19 09:44

阅读 7335

PyTorch

1、gather函数

`torch.gather(input,dim,index,out=None)`

作用：获取输入tensor中特定维度特定位置的元素

参数：

`input(tensor) --源张量 `

`dim(int) -- 索引维度`

`index(Tensor) -- 待获取元素的索引`

`out -- 输出张量`

examples:

![](https://dataset.flyai.com/2021-08-18 17-57-15 的屏幕截图.png)
![](https://dataset.flyai.com/2021-08-18 17-56-42 的屏幕截图.png)

官方文档3维张量操作说明：

![](https://dataset.flyai.com/2021-08-18 22-11-31 的屏幕截图.png)

故可知：
     `out[0][1]`对应为`input[0][index[0][1]]` ,即为`input[0][0]`，所以该位置值为1。其他位置类似。

2、scatter_函数

`scatter_(input,dim,index,src)`

作用：将src中的元素根据index中的索引取出并在指定维度上填进input

参数：

`input(tensor) --被填充张量`

`dim(int) -- 索引维度`

`index(Tensor) -- 待获取元素的索引`

`src(tensor) -- 源张量`

examples:

![](https://dataset.flyai.com/2021-08-18 17-58-25 的屏幕截图.png)
   ![](https://dataset.flyai.com/2021-08-18 17-58-17 的屏幕截图.png)
官方文档3维张量操作说明：

![](https://dataset.flyai.com/2021-08-18 22-15-01 的屏幕截图.png)
故可知：
     
	 由于out1.shape[0] > inp.shape[0]，所以只能在dim=0维度上进行填充

其中`out1[index1[0][1]][1] = inp[0][1] ,即out1[1][1] = 2`

本作品采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可，转载请附上原文出处链接和本声明。
本文链接地址：https://flyai.com/article/971

讨论

500字

表情

发送

删除确认

是否删除该条评论？

取消删除