pytorch将label转为one hot形式

label_one_hot = torch.nn.functional.one_hot(labels, self.num_classes).float().to(self.device)

https://pytorch.org/docs/stable/nn.functional.html#torch.nn.functional.one_hot

你可能感兴趣的