I need to downsample my label tensor from (n, c, W, H) to (n, c, w, h) and my tensor has type of torch.int64 because it is a label tensor. I tried to use F.interpolate(lb, size, mode = 'nearest'), but it has the result of

RuntimeError: upsample_nearest1d_forward is not implemented for type torch.cuda.LongTensor

So what is the correct way to do this?