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]
那就看是取行还是列了。
取行就是数组的 第一行和第二行
取列同理。