pytorch中获得one-hot标签的方法 label_onehot = torch.zeros(BATCH_SIZE, NUM_CLASSES).scatter_(1, label.reshape(-1,1), 1) 其中label是一个一维的tensor,数据类型为long