def METHOD_NAME(self, target):
    input = target.clone()
    input[input == -1] = 1  # replace negative label with class prediction
    input = F.one_hot(input)
    input = torch.movedim(input, 3, 1)
    return input