read mnist has now an option to return the shaped images instead of the fattened vectors