Doubt in code for sgd


why we are reshaping here?