Moved the input/output shift in the forward of the model.