print(labels.detach().numpy())
labels.unsqueeze_(-1)
print(labels.detach().numpy())

labels_onehot = torch.FloatTensor(batch_size, 10)
labels_onehot.zero_()
labels_onehot.scatter_(1, labels, 1)
print(labels_onehot.detach().numpy())