Function one_hot

models/mixmatch/mixmatch_utils.py:18–20

(targets, nClass, gpu)

def one_hot(targets, nClass, gpu):
    logits = torch.zeros(targets.size(0), nClass).cuda(gpu)
    return logits.scatter_(1, targets.unsqueeze(1), 1)


def mixup_one_target(x, y, gpu, alpha=1.0, is_bias=False):