Convert the 1-D integer tensor labels into a one-hot encoded matrix with num_classes columns: row i is all zeros except a 1 at column labels[i].
labels
num_classes
i
labels[i]
solve(...)