鹤壁河南网站建设,乘风专业建站,建立网站如何,移动互联网开发找工作Pytorch中的gather的理解和用法
这个Gather的用法花费了点时间#xff0c;我相信很多人一开始不太懂。 跟着我简单理解。
首先样例是#xff1a;
tensor([[ 3, 4, 5],[ 6, 7, 8],[ 9, 10, 11]])然后index:
[[2, 1, 0]]然后执行的代码#xff1a;
tensor_0.gather(0…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)
那我们应该找的原数组的值就是 [11]的位置。 对于索引数组中 “0”的位置是多少[0,2]对不对
假定我们替换行 即输入.gather(0, index)
那我们应该找的原数组的值就是 [02]的位置。
取列就是同理了。
还有一种情况就是单纯的数组
index :[1,2] 那就看是取行还是列了。 取行就是数组的 第一行和第二行 取列同理。
