Pytorch中的gather的理解和用法

Pytorch中的gather的理解和用法

这个Gather的用法花费了点时间，我相信很多人一开始不太懂。
跟着我简单理解。

首先样例是：

tensor([[ 3,  4,  5],
        [ 6,  7,  8],
        [ 9, 10, 11]])

然后index:

[[2, 1, 0]]

然后执行的代码：

tensor_0.gather(0, index)
tensor_0.gather(1, index)

这里的0和1分别代码行和列！

你可以理解为替换行或者替换列！

怎么替换呢？

索引数组本身数值的位置，用数值去替换。

这是索引：

[[2, 1, 0]]

对于索引数组中 “2”的位置是多少？：[0,0]对不对？

假定我们替换行 即输入.gather(0, index)

那我们应该找的原数组的值就是 [2,0]的位置。

以此类推：
对于索引数组中 “1”的位置是多少？：[0,1]对不对？

假定我们替换行 即输入.gather(0, index)

那我们应该找的原数组的值就是 [1，1]的位置。

---

对于索引数组中 “0”的位置是多少？：[0,2]对不对？

假定我们替换行 即输入.gather(0, index)

那我们应该找的原数组的值就是 [0，2]的位置。

取列就是同理了。

还有一种情况就是单纯的数组

index :[1,2]
那就看是取行还是列了。
取行就是数组的 第一行和第二行
取列同理。