Ah I see. Try the following:
import torch
input = torch.randn(3, 3)  # the 2d tensor with eleemnts
index = torch.LongTensor([1, 2])  # your hypothetical index
input[index[0], index[1]]  # gets you the element at (1, 2) of input