Not what I wanted.
train = data_utils.TensorDataset(torch.LongTensor(dataX),torch.LongTensor(dataY))
train_loader = data_utils.DataLoader(train,batch_size=512,drop_last=True)
for batch_idx,train in enumerate(train_loader):
    print(train.shape)