creating data loader to feed data to the network

data-loading.py

import torch
from torch.utils.data import TensorDataset
from torch.utils.data import DataLoader

# Converting from numpy arrays to torch tensors
x_train = torch.from_numpy(train_set_std[:,:-1]).float()
y_train = torch.from_numpy(train_set_std[:,-1]).long()

# Creating tensor-datasets
train_ds = TensorDataset(x_train, y_train)

# Creating data-loaders
batch_size = 32
train_dl = DataLoader(train_ds, batch_size=batch_size)