[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"))