From the documentation, to create a TensorDataset I just need the x and y values, and pass that into the TensorDataset? Does it require the y values be in a one-hot encoded tensor (so n x n_classes) or can it be just an n x 1 tensor?