Solved by Linus_Tomasulo in post #2 
              
              
                Solved with: 
coordinates = coordinates.unsqueeze(1)
features = torch.nn.functional.grid_sample(features, coordinates, align_corners=True)
features = features.squeeze(2)