gather

1、作用是将input中的数据按dim和index取出
2、
out[i][j][k] = input[index[i][j][k]][j][k]  # if dim == 0
out[i][j][k] = input[i][index[i][j][k]][k]  # if dim == 1
out[i][j][k] = input[i][j][index[i][j][k]]  # if dim == 2
3、
dim = 0表示的是对纵向的操作也就是列的操作

　

dim =1表示是对横向的操作也就是行的操作