You could maybe use torchvision.transforms.Normalize or write the normalization manually.
E.g. if you want to standardize the input you could subtract the mean and divide by the stddev.
Feel free to add more information about your use case, if you get stuck.