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表示是对横向的操作也就是行的操作
