Gathering values of a list of matrix indices/values into a matrix with torch.gather()

Question

I have a numpy array of shape (80,000, 3). Each row of this array has entries [j,i,s] which correspond the the value s at index [i,j]