在学习神经网络的路上遇到了gather,看了两天都没整明白,也看到了很多文章,但是一直都迷迷糊糊的,这是自己突然看原文档解释后,慢慢摸索明白,如果有问题请各位指教。
原解释文档用的3维数组举例
out[i][j][k] = input[index[i][j][k]] [j][k] when dim=0
out[i][j][k] = input[i] [index[i][j][k]] [k] when dim=1
out[i][j][k] = input[i][j] [index[i][j][k]] when dim=2