I have a question regarding tensor slices. My variables vtf and a have torch.FloatTensor of size 5x25 and torch.LongTensor of size 5x5 respectively. When I do vtf[:,a[0,:]], it gives me torch.FloatTensor of size 5x5. However if I do vtf[0,a[0,:]]  then I received TypeError: indexing a tensor with an object of type torch.LongTensor. The only supported types are integers, slices, numpy scalars and torch.LongTensor or torch.ByteTensor as the only argument.
Please let me know if I am doing something wrong