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