Required a `normalize` function for a custom transform in `data_loader`

[SaashaJoshi]

Normalization must be done in desired ranges by the user. To achieve this, a custom torchvision.transforms input in data_loader is required. This can be solved by defining a normalize function that takes in user inputs.