Prepare input data

Typically, we adjust input to have 0 mean and 1 variance for numerical stability and for the math.