How to sort a 3d tensor by coordinates in last dimension (pytorch)

Question

I have a tensor with shape [bn, k, 2]. The last dimension are coordinates and I want each batch to be sorted independently depending on the y coordinate (