6ddf825a4aa1bcf3a35bba8d36bc433fe5ebaa39,torchsample/utils.py,,th_nearest_interp_3d,#Any#Any#,285

Before Change

    // gather image values at coordinates
    mapped_vals = torch.stack([th_gather_nd(input[i], coords[i])
                    for i in range(input.size(0))], 0)

    return mapped_vals.view_as(input)

After Change

    input_flat = th_flatten(input)

    mapped_vals = input_flat[idx.long()]

    return mapped_vals.view_as(input)

Instances