You need to consider that each of item from the second dim (492) also has 5 items. How you
want them to be sorted?
In general there is no direct way to do it. but you can store the column you want in a new array then
use torch.sort() with which you will get the indices of the sorted items. with that you can use torch.index_select() and get a sorted tensor. but you need to be careful of the other dimensions  not
just the batch and the one through which you want to sort.