def get_swiss_roll_loader(n_samples=1000):
    (X, _) = get_swiss_roll_data(n_samples)
    dataset = torch.utils.data.dataset.TensorDataset(torch.FloatTensor(X), torch.FloatTensor(_))
    loader = torch.utils.data.dataloader.DataLoader(dataset, batch_size=100, shuffle=True)
    return loader
def get_swiss_roll_loader(n_samples=1000):
    (X, _) = get_swiss_roll_data(n_samples)
    dataset = torch.utils.data.dataset.TensorDataset(torch.FloatTensor(X), torch.FloatTensor(_))
    loader = torch.utils.data.dataloader.DataLoader(dataset, batch_size=100, shuffle=True)
    return loader