In session no.2 we extended the ‘nn.module’ class for reshaping the image tensor. But in session no.3, we have just used ‘.reshape’. Why haven’t we extended ‘nn.module’ class ? Can we do similar thing is session no.2 as well instead of extending the ‘nn.module’ ?