15c7935788534423c3b22626fb1a6978a59878b6,torchsample/utils.py,,th_nearest_interp_2d,#Any#Any#,112

Before Change

    input_flat = th_c_flatten(input)

    mapped_vals = torch.stack([input_flat[i][idx] 
                    for i in range(input.size(0))], 0)

    return mapped_vals.view_as(input)

After Change

    coords[:,0] = torch.clamp(coords[:,0], 0, input.size(1)-1).round()
    coords[:,1] = torch.clamp(coords[:,1], 0, input.size(2)-1).round()

    stride = torch.LongTensor(input.stride())[1:].float()
    idx = coords.mv(stride).long()

    input_flat = th_flatten(input)

Instances