Given the following 2D tensor whit shape [N,E]:
#N=3, E=3
r1 = torch.tensor([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

how to swap the r1 “rows” to get results like the one shown below:
r1 = torch.tensor([
    [1, 2, 3],
    [7, 8, 9],
    [4, 5, 6]
    
])

Note: any permutation between the “rows” would be valid.