X 为 features, y 为 label,
from torch.utils import data
dataset = data.TensorDataset((X, y))
train_iter = data.DataLoader(dataset, batch_size, shuffle) #生成一个迭代器
X 为 features, y 为 label,
from torch.utils import data
dataset = data.TensorDataset((X, y))
train_iter = data.DataLoader(dataset, batch_size, shuffle) #生成一个迭代器