Say i have a float tensor A with dim[n][n], and 2 int tensor X,Y with dim[N].
each value in X,Y is a int between [0,n-1]
now i want to create a tensor B with dim[N,N], satisfying B[i][j]=A[X[i]][Y[j]]
help me, plz