Added a new argumnet to deep belief nets so that you can keep the initial shape of the dataset