Generic caller for all the transformations responsible to make model reshape-able by batch dimension.