I am doing classification task. I want my X as four dimension and my y as 1 dimension but getting error while I am trying to stack it together.
train = torch.utils.data.TensorDataset(X_train, y_train), the error is coming from this line.