pytorch各种操作

pytorch为了加快数据提取，定义建立train_data,train_loader

train_data = TensorDataset(torch.tensor(train_data, dtype=torch.int))

train_loader = DataLoader(train_data, batch_size= batch_size,shuffle=True, drop_last=True)