from torch.utils.data import [DataLoader](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader "torch.utils.data.DataLoader")

[train_ds](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset") = [TensorDataset](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset")([x_train](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor"), [y_train](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor "torch.Tensor"))
train_dl = [DataLoader](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.DataLoader "torch.utils.data.DataLoader")([train_ds](https://docs.pytorch.org/docs/stable/data.html#torch.utils.data.TensorDataset "torch.utils.data.TensorDataset"), batch_size=bs)