Permuting a data set

Hi, I’m generating a data set of size [batch_size, 2], where each row entry is a 2D coordinate from 0-1 (first column is x-coordinate, and second is y-coordinate). Is there a way to quickly rearrange this data set such that the first entry is always at the furthest right corner (e.g. closest to (1,1))?